AssertionError: MaskedFill can't differentiate the mask about stargan HOT 5 CLOSED

change lines of the method as follows:

x = (x >= 0.5).float()

from stargan.

Yes!Thank you so much. In detail,
two lines in solver.py change from
" x[x >= 0.5] = 1
x[x < 0.5] = 0"
TO:
"x = (x >= 0.5).float()
x = (x < 0.5).float()"

from stargan.

@mrlzla thank you! it's a bug of pytorch in version 0.3. when i switch to pytorch 0.2, everything runs well!!!!

from stargan.

@mrlzla Thanks a lot. I changed the code of this repo :)
@SummerHuiZhang You should not add the code "x = (x < 0.5).float()".

from stargan.

well, you could just use torch.clamp(x , 0, 0.5) to limit the x in the range (0, 0.5)

from stargan.