0
0

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 3 years have passed since last update.

4GBのGPUメモリで、CycleGANを動かしてみた。

Posted at

はじめに

https://github.com/eriklindernoren/PyTorch-GAN
に、沢山のGANがある。

このgithubのコードは、非常に簡単に動作させることができる。

CycleGANを動かしてみた

以下が基本コマンド。

python3 cyclegan.py --dataset_name monet2photo

メモリが足りないと出る。
通常、まず、batch_sizeを小さくしたりするが、
既に、1になっているようなので、
仕方なく、勘で、画像サイズを小さく指定したら、動いてくれた。

--img_height 128 --img_width 128

動作結果

非力なPCなので、epochを極端に小さくした。2にした。
それでも、1日弱時間がかかった。
上段がmonet、下段がそれを入力とした実写(化)、のはず。
まだ、どっちがどっちというレベルではありますが。。。
別途、もう少し、わかりやすいサンプルでやってみます。

image.png

コメント

特にありません。
このgithubは、お勧めーーーー。

0
0
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
0
0

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?