Minor fixes - Githubissues

aitorzip / PyTorch-CycleGAN

A clean and readable Pytorch implementation of CycleGAN

https://arxiv.org/abs/1703.10593

GNU General Public License v3.0

1.19k stars 283 forks source link

Open tanvach opened 6 years ago

tanvach commented 6 years ago

Fixes two small issues I came across before I could get the script to run:

Pytorch now treats true scalar (dimension [1]) differently to a singleton (dimension [1x1]). This broke the loss computation at loss_GAN_A2B = criterion_GAN(pred_fake, target_real) step. Fixed by making target_real and target_fake tensors.
Method to start Visdom is now updated

@aitorzip

Pandinosaurus commented 6 years ago

Thank you @tanvach ! Your patch also does the job on my setup.