Increase number of saved models

rien333 · Jun 6, 2018 · 781a713 · 781a713
1 parent 59a7c83
commit 781a713
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/conv_vae_pytorch.py b/conv_vae_pytorch.py
@@ -628,7 +628,7 @@ def test(epoch, loader):
 
 def train_routine(epochs, train_loader, test_loader, optimizer, scheduler, reset=120, start_epoch=0):
     # This could/should be a dictionary
-    best_models = [("", 100000000000)]*3
+    best_models = [("", 100000000000)]*5
 
     for epoch in range(start_epoch, epochs + 1):
         train(epoch, train_loader, optimizer)