From 4ef813555c460a2ff81dd56db19fcafff84c13a8 Mon Sep 17 00:00:00 2001
From: Shen-Chenhui <e0052680@u.nus.edu>
Date: Mon, 8 Apr 2024 17:22:44 +0800
Subject: [PATCH] debug

---
 scripts/inference-debug.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/inference-debug.py b/scripts/inference-debug.py
index b75c2a5..179cd89 100644
--- a/scripts/inference-debug.py
+++ b/scripts/inference-debug.py
@@ -121,7 +121,7 @@ def main():
     # latent_size = vae.get_latent_size(input_size)
 
     # 3.2. move to device & eval
-    vae = vae.to(device, dtype)
+    vae = vae.to(device, dtype).eval()
 
     # # 4.5. setup optimizer
     # optimizer = HybridAdam(