fix(trainer): missing check

webpolis · Aug 12, 2023 · 9706373 · 9706373
1 parent 25efb3d
commit 9706373
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/tools/trainer.py b/src/tools/trainer.py
@@ -336,6 +336,7 @@ def on_train_epoch_end(self, trainer, pl_module):
         ) else
         args.vae_emb[1] if
         (
+            args.vae_emb != None and
             len(args.vae_emb) > 1 and VAE_MODE == 'train' and os.path.isfile(
                 args.vae_emb[1])
         ) else