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

Data Architect

@ University of Texas at Austin | Austin, TX

Data ETL Engineer

@ University of Texas at Austin | Austin, TX

Lead GNSS Data Scientist

@ Lurra Systems | Melbourne

Senior Machine Learning Engineer (MLOps)

@ Promaton | Remote, Europe

Business Data Scientist, gTech Ads

@ Google | Mexico City, CDMX, Mexico

Lead, Data Analytics Operations

@ Zocdoc | Pune, Maharashtra, India