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__(
9494
9595 if self ._need_data :
9696 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 ,
113113 )
114114
115115 self ._lazy_dataloader_iter = None
You can’t perform that action at this time.
0 commit comments