worker_init_fn in train_cross_teaching_between_cnn_transformer_2D.py

HiLab-git / SSL4MIS

Semi Supervised Learning for Medical Image Segmentation, a collection of literature reviews and code implementations.

MIT License

2.13k stars 381 forks source link

worker_init_fn in train_cross_teaching_between_cnn_transformer_2D.py #67

Open Padmini588 opened 1 year ago

Padmini588 commented 1 year ago

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)