Update train_latent.py

Fix latent distribution
mlpc-ucsd · May 17, 2024 · 7abc7dd · 7abc7dd
1 parent 3aa9260
commit 7abc7dd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/train_latent.py b/train_latent.py
@@ -20,6 +20,7 @@
     gpus = [0]
     conf.sample_size = 1
     conf.semantic_enc = args.semantic_enc
+    conf.latent_znormalize = False
     conf.pretrain = PretrainConfig(
         name='72M',
         path=model_path,