Last active
October 2, 2019 11:59
-
-
Save lucasc896/017088d990980c43adec0af2dcb1c45c to your computer and use it in GitHub Desktop.
1Konny/Beta-VAE bugfix
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| diff --git a/solver.py b/solver.py | |
| index 7d65766..57ab586 100644 | |
| --- a/solver.py | |
| +++ b/solver.py | |
| @@ -179,7 +179,7 @@ class Solver(object): | |
| if self.global_iter%self.display_step == 0: | |
| pbar.write('[{}] recon_loss:{:.3f} total_kld:{:.3f} mean_kld:{:.3f}'.format( | |
| - self.global_iter, recon_loss.data[0], total_kld.data[0], mean_kld.data[0])) | |
| + self.global_iter, recon_loss.item(), total_kld.item(), mean_kld.item())) | |
| var = logvar.exp().mean(0).data | |
| var_str = '' |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Thanks! Will try it!