fix solver init problem when data iter is not given

dmlc · May 13, 2017 · 77896e5 · 77896e5
1 parent 0e571ac
commit 77896e5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/minpy/nn/__init__.py b/minpy/nn/__init__.py
@@ -1,2 +0,0 @@
-import os
-os.environ['MXNET_ENGINE_TYPE'] = 'NaiveEngine'

diff --git a/minpy/nn/solver.py b/minpy/nn/solver.py
@@ -156,8 +156,10 @@ def _reset(self):
             self.init_configs[p] = init_config
 
     def _reset_data_iterators(self):
-        self.train_dataiter.reset()
-        self.test_dataiter.reset()
+        if self.train_dataiter:
+            self.train_dataiter.reset()
+        if self.test_dataiter:
+            self.test_dataiter.reset()
 
     def _step(self, batch):
         """