jj0mst

Why recon_criterion is not implemented with PyTorch L1Loss()?

1

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...

I recently tried to use the new BCHW functions with my network, since i always use that layout and it simplifies my code. I noticed that all the gradients of...

Hi, can you share the weights of the network after training? They would be useful for running simple inference or plugging the trained network into other systems. I saw that...

How to avoid issues with dictionaries?

1

Hello, I found the repo and blog post very interesting and useful, especially the tensor serialization utility. I've encountered similar problems with RAM usage, but in my case I have...