From: François Fleuret Date: Fri, 1 Mar 2024 22:33:21 +0000 (+0100) Subject: Update. X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=commitdiff_plain;h=39ce2c3b3e9f4a0d4da6cb7eb84d15cb03c55ff4 Update. --- 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()