worker_init_fn in train_cross_teaching_between_cnn_transformer_2D.py
Padmini588 opened this issue · comments
Hello ,
In the below line i do no not see the initialisation of worker_init_fn in the code available in the file train_cross_teaching_between_cnn_transformer_2D.py. Please can you help me with this.
trainloader = DataLoader(db_train, batch_sampler=batch_sampler,
num_workers=4, pin_memory=True, worker_init_fn=worker_init_fn)