UGATIT-pytorch icon indicating copy to clipboard operation
UGATIT-pytorch copied to clipboard

use light mode G not work at all

Open HUSTLX opened this issue 4 years ago • 7 comments

just like this, which D has a very small loss image

HUSTLX avatar Aug 20 '19 08:08 HUSTLX

may need to train more steps

Frizy-up avatar Aug 21 '19 01:08 Frizy-up

may need to train more steps

already, it dose not work

HUSTLX avatar Aug 21 '19 03:08 HUSTLX

I also met such case

saturosfz avatar Aug 22 '19 03:08 saturosfz

+1, I already trained more than 100000 steps but still got blank outputs.

cfanyyx avatar Aug 22 '19 04:08 cfanyyx

+1, Using light mode gets similar results.

xiaosean avatar Aug 22 '19 05:08 xiaosean

It seems the input channel after adaptive_avg_pooling in light mode can't be set too small.

if self.light:
    FC = [nn.Linear(ngf * mult, ngf * mult, bias=False),  
          nn.ReLU(True),  
          nn.Linear(ngf * mult, ngf * mult, bias=False),  
          nn.ReLU(True)]

if I change the first ngf * mult to a larger one, then the output become normal.

cfanyyx avatar Aug 22 '19 08:08 cfanyyx

I try with this command python main.py --dataset selfie2anime --light True --resume True --save_freq 10000 --benchmark_flag True and the following is my result at 100,000 epoch. btw I continue the training start from 10000 (it is just the save network testing before).

B2A is terrible...

jayzhan211 avatar Feb 01 '20 23:02 jayzhan211