I was training my model using Yolo v3
When I set my input size = 416, I can train my model with batch size = 9 without any errors.
However, when I decrease my input size to 320, I  ran into Cuda memory error even when my batch size = 7.
I found this particularly strange, has anyone encountered anything similar before?
Error happened when executing loss.backward()
Thank you in advance for helping me!