WindowsでAIプログラミング

LightWeight-GAN:0.20.0にバージョンアップ、Dual Contrastive Loss やってみました

はじめに

『LightWeight-GAN』が『0.20.0』にバージョンアップしました。
新たに『Dual Contrastive Loss』というオプションが追加されました。
というわけで学習を比較してみました。

新規追加コマンドオプション

Dual Contrastive Loss

A recent paper has proposed that a novel contrastive loss between the real and fake logits can improve quality slightly over the default hinge loss.

You can use this with one extra flag as follows

学習条件

顔の画像1024×1024を約1万枚。
オーギュメンテーションは行っていません。
『Google Colaboratory Pro』を使用して学習を行っています。

結果

学習経過

100itr

LightWeight-GAN 0.17
LightWeight-GAN 0.20 –dual_contrast_loss

500itr

LightWeight-GAN 0.17
LightWeight-GAN 0.20 –dual_contrast_loss

1000itr

LightWeight-GAN 0.17
LightWeight-GAN 0.20 –dual_contrast_loss

5000itr

LightWeight-GAN 0.17
LightWeight-GAN 0.20 –dual_contrast_loss

100000itr

LightWeight-GAN 0.17
LightWeight-GAN 0.20 –dual_contrast_loss

200000itr

LightWeight-GAN 0.17
LightWeight-GAN 0.20 –dual_contrast_loss

300000itr

LightWeight-GAN 0.17
LightWeight-GAN 0.20 –dual_contrast_loss

400000itr

LightWeight-GAN 0.17
LightWeight-GAN 0.20 –dual_contrast_loss

Interpolation

LightWeight-GAN 0.17

LightWeight-GAN 0.20 –dual_contrast_loss

学習開始初期は、『–dual_contrast_loss』の方が、学習が速く思えましたが、
後はあまり変わりないような感じでした。

補足

同じ学習条件でも、『0.17』で作った学習モデルに『0.20』で追加学習は、できませんでした。

この記事を書いた人
化学系で博士号を取得したが、 あるとき、これからの時代はプログラミング!、と目覚める。 pythonを用いてデータ解析や機械学習に没頭。 最近は、Pytorchで作ったONNXモデルを、Nuxt3にのせたWebサービスの開発、 ChatGPT や Stable Diffusion に没頭中☆('ω')☆
SNSでフォローする

コメントを残す

メールアドレスが公開されることはありません。 が付いている欄は必須項目です

CAPTCHA