Pytorch DataLoader do not have a reset() function.

RaduAlexandru · Jan 21, 2021 · 19dc805 · 19dc805
1 parent a2ab360
commit 19dc805
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/latticenet_py/callbacks/state_callback.py b/latticenet_py/callbacks/state_callback.py
@@ -47,4 +47,8 @@ def phase_started(self, phase, **kwargs):
         phase.samples_processed_this_epoch=0
 
     def phase_ended(self, phase, **kwargs):
-        phase.loader.reset()
+
+        if(type(phase.loader) == torch.utils.data.DataLoader): # pytorchs dataloder has no reset 
+            pass
+        else:
+            phase.loader.reset()