removegc.diff

commit 3948543212d3d8286a687ef7783783bae720f65f
Author: Sergey Zagoruyko <[email protected]>
Date:   Mon Nov 21 18:01:56 2016 +0100

    do gc only on val

diff --git a/dataloader.lua b/dataloader.lua
index a995bff..b69482a 100644
--- a/dataloader.lua
+++ b/dataloader.lua
@@ -79,7 +79,9 @@ function DataLoader:run()
                   batch[i]:copy(input)
                   target[i] = sample.target
                end
-               collectgarbage()
+               if _G.dataset.split == 'val' then
+                  collectgarbage()
+               end
                return {
                   input = batch:view(sz * nCrops, table.unpack(imageSize)),
                   target = target,