Training with a Single GPU - CUDA out of memory.

Hey,

I have been trying to run the repo over a custom dataset for a while now. I believe that I have the custom dataset prepared accordingly after some hassling.

However, now I am stuck with getting out of CUDA memory. Could you help me find out what to change in the configs that I can reduce the video memory occupation during training (e.g. mini-batching, etc.)?

I am currently using 2 samples per GPU with number of GPUs = 1 using the bash code below (just to get the repo working),

for FOLD in 1;
do
  bash tools/dist_train_partially.sh semi ${FOLD} 5 1
done

Thanks

[MendelXu]

I have tried to do this and I think the only solution is to decrease the sizes of input images in your case.

[MendelXu]

https://github.com/microsoft/SoftTeacher/blob/bef9a256e5c920723280146fc66b82629b3ee9d4/configs/soft_teacher/base.py#L30 https://github.com/microsoft/SoftTeacher/blob/bef9a256e5c920723280146fc66b82629b3ee9d4/configs/soft_teacher/base.py#L80 https://github.com/microsoft/SoftTeacher/blob/bef9a256e5c920723280146fc66b82629b3ee9d4/configs/soft_teacher/base.py#L153 Change these lines.