MUNIT
MUNIT copied to clipboard
Why recon_criterion is not implemented with PyTorch L1Loss()?
I dont' understand why you use this line as reconstruction loss: https://github.com/NVlabs/MUNIT/blob/4837f034fba1a49d6d6675e7eb954caefd698375/trainer.py#L54
I came from the UNIT repository where it is implemented with the official PyTorch L1Loss and saw that the reconstruction for the content features explodes (e.g. loss_gen_recon_c_a is 50 millions at the first iteration). Do you know why this happens? Aren't they equivalent?
I am not the author, but I think that it's because L1Loss can't handle 2 Tensors with requires_grad=True I faced the same problem