address comments

apache · Sep 16, 2019 · 76ca2cc · 76ca2cc
1 parent bb2187e
commit 76ca2cc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/mxnet/gluon/data/dataloader.py b/python/mxnet/gluon/data/dataloader.py
@@ -572,6 +572,7 @@ def __init__(self, dataset, batch_size=None, shuffle=False, sampler=None,
         self._pin_device_id = pin_device_id
         self._thread_pool = thread_pool
         self._timeout = timeout
+        assert timeout > 0, "timeout must be positive, given {}".format(timeout)
 
         if batch_sampler is None:
             if batch_size is None: