NVlabs / FUNIT

Translate images to unseen domains in the test time with few example images.

Home Page:https://nvlabs.github.io/FUNIT/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

The training code does not work well...

UdonDa opened this issue · comments

commented

The result which is generated by the model is trained during 400 iterations.

I think it looks that the models fail to learn.
I modify inplace=False in nn.ReLU and nn.LeakyReLU in blocks.py. following this issue #23

gen_train_00000400_02

Thanks.

It takes a while for the model to generate some nice pictures. I trained with batch size of 16 pictures, and the pictures become reasonable after something like 20K iterations. It is possible to stabilize the training by reducing the learning rate, but then the final results look worse.

commented

gen_train_00050500_03

I can get the clear samples! thanks!