From 39ce2c3b3e9f4a0d4da6cb7eb84d15cb03c55ff4 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Fri, 1 Mar 2024 23:33:21 +0100 Subject: [PATCH] Update. --- tiny_vae.py | 1 + 1 file changed, 1 insertion(+) 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() -- 2.39.5