DRNET icon indicating copy to clipboard operation
DRNET copied to clipboard

Why the rec_loss returned from function train_main_network() is divided by args.batch_size in main.py?

Open Dawn90 opened this issue 7 years ago • 0 comments

  1. Why the rec_loss returned from function train_main_network() is divided by args.batch_size in main.py? As we know, nn.MSELoss() already returns the mean loss by default. line 183: return sim_loss.data/args.batch_size, rec_loss.data/args.batch_size

  2. Are the default parameters just right for the MNIST dataset? I run python main.py for 4500 iterations with the default parameters, and the reconstruction loss had converged to 0.000126. But the visualization of the reconstructed image is almost a black picture. I'm really confused where I make a mistake. Could you give me any instructions?

Thank you verrrrrry much! Best, Dong

Dawn90 avatar Jul 13 '18 18:07 Dawn90