Add collectgarbage in main to avoid PANIC errors

fmassa · May 26, 2015 · 7b34a56 · 7b34a56
1 parent 6a5dcbf
commit 7b34a56
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/main.lua b/main.lua
@@ -66,7 +66,9 @@ for i=1,opt.num_iter do
   trainer:train(inputs,targets)
   print('==> Training Error: '..trainer.fx[i])
   print(trainer.confusion)
-
+
+  collectgarbage() 
+
   err = validator:validate(criterion)
   print('==> Validation Error: '..err)
   table.insert(val_err,err)