mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Change the epoch runner to use the data_loader from attributes rather than args. (#483)

pull/488/head
su 2020-08-12 22:09:07 +08:00 committed by GitHub
parent e92f826abc
commit 17e4732c49
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mmcv/runner/epoch_based_runner.py
View File

@ -21,10 +21,10 @@ class EpochBasedRunner(BaseRunner):
self.model.train() self.model.train()
self.mode = 'train' self.mode = 'train'
self.data_loader = data_loader self.data_loader = data_loader
self._max_iters = self._max_epochs * len(data_loader) self._max_iters = self._max_epochs * len(self.data_loader)
self.call_hook('before_train_epoch') self.call_hook('before_train_epoch')
time.sleep(2) # Prevent possible deadlock during epoch transition time.sleep(2) # Prevent possible deadlock during epoch transition
for i, data_batch in enumerate(data_loader): for i, data_batch in enumerate(self.data_loader):
self._inner_iter = i self._inner_iter = i
self.call_hook('before_train_iter') self.call_hook('before_train_iter')
if self.batch_processor is None: if self.batch_processor is None:
@ -52,7 +52,7 @@ class EpochBasedRunner(BaseRunner):
self.data_loader = data_loader self.data_loader = data_loader
self.call_hook('before_val_epoch') self.call_hook('before_val_epoch')
time.sleep(2) # Prevent possible deadlock during epoch transition time.sleep(2) # Prevent possible deadlock during epoch transition
for i, data_batch in enumerate(data_loader): for i, data_batch in enumerate(self.data_loader):
self._inner_iter = i self._inner_iter = i
self.call_hook('before_val_iter') self.call_hook('before_val_iter')
with torch.no_grad(): with torch.no_grad():