Remove use of memonger

apache · Aug 12, 2018 · f9f8e3f · f9f8e3f
1 parent 11a8f64
commit f9f8e3f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/example/profiler/profiler_executor.py b/example/profiler/profiler_executor.py
@@ -84,9 +84,7 @@ def get_symbol():
     net  = mx.symbol.SoftmaxOutput(data=fc5 , name='softmax')
     return net, [('data', (128, 1, 28, 28))], [('softmax_label', (128, ))]
 
-def get_module(ctx, sym, provide_data, provide_label, batch_size=None, is_train=True, use_memonger=False):
-    if use_memonger:
-        sym = search_plan(sym, data=data_shapes)
+def get_module(ctx, sym, provide_data, provide_label, batch_size=None, is_train=True):
     mod = mx.mod.Module(symbol=sym,
                         data_names=[name for name, _ in provide_data],
                         label_names=[name for name, _ in provide_label],