search

fhaghighi / DiRA

Official PyTorch Implementation for DiRA: Discriminative, Restorative, and Adversarial Learning for Self-supervised Medical Image Analysis - CVPR 2022
Other
98 stars 9 forks source link

Question about training details of ChestX-ray14 #2

Open Astrostellar opened 2 years ago

Astrostellar commented 2 years ago

I think it is a great job for medical image analysis! However, I have some questions about the details of training 2D images. ①You did a warming up and then jointly training, which totally took up to 800 epoches as shown in your paper. Could you please tell me how many epoches it takes for warming up and how many epoches for jointly training? ②In the jointly training stage, you set the batch size to 16 instead of 256 in the repository, but it seems not to be mentioned in your paper. But I found it take little gpu memory when I set batch size to 16 and work well for batch size of 256. Is there any reason to set the batch size smaller? What happened if larger batch size like 256 is used during jointly training? Thanks in advance for your time!

TIanCat commented 1 year ago

I have the same question about training details. And I want to know how many epochs the original CL mathods(moco v2/simsiam/barlow twins) was trained? is it 800 epochs?