add back memory_optimize

dzhwinter · Mar 28, 2018 · 61ee82e · 61ee82e
1 parent 70f13e7
commit 61ee82e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fluid/SE-ResNeXt-152/train.py b/fluid/SE-ResNeXt-152/train.py
@@ -212,7 +212,7 @@ def train():
         regularization=fluid.regularizer.L2Decay(1e-4))
     opts = optimizer.minimize(avg_cost)
 
-    # fluid.memory_optimize(fluid.default_main_program())
+    fluid.memory_optimize(fluid.default_main_program())
 
     place = fluid.CUDAPlace(0)
     # place = fluid.CPUPlace()