HiLab-git / SSL4MIS

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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)