X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=blobdiff_plain;f=tiny_vae.py;fp=tiny_vae.py;h=0895830a13d123a511c12c6b37e6ff414936f39e;hp=d33cc4bbf2cec2833f66d8c85fa0ac5d3063b4e2;hb=39ce2c3b3e9f4a0d4da6cb7eb84d15cb03c55ff4;hpb=9529aef0fdc1d1810e3c331a01fdbe640669ecfc diff --git a/tiny_vae.py b/tiny_vae.py index d33cc4b..0895830 100755 --- a/tiny_vae.py +++ b/tiny_vae.py @@ -47,6 +47,7 @@ parser.add_argument("--no_dkl", action="store_true") # With that option, do not follow the setup of the original VAE paper # of forcing the variance of X|Z to 1 during training and to 0 for # sampling, but optimize and use the variance. + parser.add_argument("--no_hacks", action="store_true") args = parser.parse_args()