Manually fully initialize the dataset in runner when building dataloader (#190)
* full init the dataset when building data loader * resolve commentspull/198/head
parent
4f5c0cd273
commit
fb7d8ccd6b
|
@ -906,6 +906,8 @@ class Runner:
|
|||
dataset_cfg = dataloader_cfg.pop('dataset')
|
||||
if isinstance(dataset_cfg, dict):
|
||||
dataset = DATASETS.build(dataset_cfg)
|
||||
if hasattr(dataset, 'full_init'):
|
||||
dataset.full_init()
|
||||
else:
|
||||
# fallback to raise error in dataloader
|
||||
# if `dataset_cfg` is not a valid type
|
||||
|
|
Loading…
Reference in New Issue