About Traning

Open lgmtxl opened this issue 5 months ago • 2 comments

Thank you for your excellent work. I'd like to ask how many GPUs you used for training the model and how long the training process took?

Jul 28 '25 09:07 lgmtxl

I have a 4090 GPU and when I train with BS=1 it gives out of memory error

Someone in the other comments section said for BS=1 training one needs minimum A100 GPU with 80G memory

Were you able to train ?

Aug 14 '25 10:08 ManuBN786

I made some modifications to gen_crop_data.py to create smaller patches and now I'm able to train on my 4090 GPU

Aug 18 '25 09:08 ManuBN786