diff options
Diffstat (limited to 'data/recursive_dataset.py')
| -rw-r--r-- | data/recursive_dataset.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/data/recursive_dataset.py b/data/recursive_dataset.py index 16f30fd..7a6d875 100644 --- a/data/recursive_dataset.py +++ b/data/recursive_dataset.py @@ -4,7 +4,7 @@ from data.base_dataset import BaseDataset, get_transform from data.image_folder import make_dataset from PIL import Image import random -# import gevent +import gevent import time class RecursiveDataset(BaseDataset): @@ -29,7 +29,7 @@ class RecursiveDataset(BaseDataset): # print('not there yet... {}'.format(A_path)) while not os.path.exists(A_path): # print('sleeping for {}'.format(self.opt.poll_delay)) - time.sleep(self.opt.poll_delay) + gevent.sleep(self.opt.poll_delay) # print("got {}".format(A_path)) A_img = Image.open(A_path).convert('RGB') A = self.transform(A_img) |