From 76ca2ccdfb0846f8aa23f506875d214e9f1e4085 Mon Sep 17 00:00:00 2001 From: "Joshua Z. Zhang" Date: Wed, 11 Sep 2019 17:30:26 -0700 Subject: [PATCH] address comments --- python/mxnet/gluon/data/dataloader.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/mxnet/gluon/data/dataloader.py b/python/mxnet/gluon/data/dataloader.py index 10de86099d04..68412267da6b 100644 --- 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: