A problem for training

[neil45853584]

Hello! The following problems occurred during my training. No response for a long time, about a few hours. What caused this?

[neil45853584]

Generally speaking, how many gigabytes of graphic memory can meet training requirements?

[ga1i13o]

GPU memory requirements can wildly vary based on the parameters that you specify. To reduce the memory footprint, you can:

• use a lighter backbone ( in the picture you are using a ResNet-50; you can use a much lighter ResNet-18)
• reduce image resolution ; e.g. we show that using 80% of original resolution ( = 384,512 ) you don't lose performances
• reduce batch size from 4 to 2

With these little tweaks you can fit into as little as 2-3 GBs of memory

[neil45853584]

Thank you! @ga1i13o I tried as you said, and it does work properly now.