Update.

diff --git a/main.py b/main.py
index e4bb494..e13c148 100755 (executable)
--- a/main.py
+++ b/main.py
@@ -947,7 +947,7 @@ def test_ae(local_device=main_device):
     model.to(local_device).train()
     optimizer_to(model.optimizer, local_device)
 
-    nb_iterations = 5
+    nb_iterations = 10
 
     for n_epoch in range(args.nb_epochs):
         # ----------------------