June 30, 2023, 8:28 a.m. | /u/Banntu

Deep Learning www.reddit.com

I'm training a keypoint-rcnn-resnet50 with pytorch on a RTX 3060 12GB. Even with batch-size 8 the gpu runs out of memory right from the beginning.
What confuses me is that the image size is only 1280x720px. Is the overhead just normally that high or could there be something wrong with my code?

batch-size deeplearning gpu image memory normally pytorch resnet50 rtx rtx 3060 something training

Senior Machine Learning Engineer

@ GPTZero | Toronto, Canada

ML/AI Engineer / NLP Expert - Custom LLM Development (x/f/m)

@ HelloBetter | Remote

Doctoral Researcher (m/f/div) in Automated Processing of Bioimages

@ Leibniz Institute for Natural Product Research and Infection Biology (Leibniz-HKI) | Jena

Seeking Developers and Engineers for AI T-Shirt Generator Project

@ Chevon Hicks | Remote

Principal Data Architect - Azure & Big Data

@ MGM Resorts International | Home Office - US, NV

GN SONG MT Market Research Data Analyst 11

@ Accenture | Bengaluru, BDC7A