File tree Expand file tree Collapse file tree 1 file changed +16
-16
lines changed Expand file tree Collapse file tree 1 file changed +16
-16
lines changed Original file line number Diff line number Diff line change @@ -94,22 +94,22 @@ def __init__(
94
94
95
95
if self ._need_data :
96
96
self ._dataloader = paddle .io .DataLoader (
97
- dataset ,
98
- feed_list ,
99
- places ,
100
- return_list ,
101
- batch_sampler ,
102
- batch_size ,
103
- shuffle ,
104
- drop_last ,
105
- collate_fn ,
106
- num_workers ,
107
- use_buffer_reader ,
108
- prefetch_factor ,
109
- use_shared_memory ,
110
- timeout ,
111
- worker_init_fn ,
112
- persistent_workers ,
97
+ dataset = dataset ,
98
+ feed_list = feed_list ,
99
+ places = places ,
100
+ return_list = return_list ,
101
+ batch_sampler = batch_sampler ,
102
+ batch_size = batch_size ,
103
+ shuffle = shuffle ,
104
+ drop_last = drop_last ,
105
+ collate_fn = collate_fn ,
106
+ num_workers = num_workers ,
107
+ use_buffer_reader = use_buffer_reader ,
108
+ prefetch_factor = prefetch_factor ,
109
+ use_shared_memory = use_shared_memory ,
110
+ timeout = timeout ,
111
+ worker_init_fn = worker_init_fn ,
112
+ persistent_workers = persistent_workers ,
113
113
)
114
114
115
115
self ._lazy_dataloader_iter = None
You can’t perform that action at this time.
0 commit comments