search

liruilong940607 / Pose2Seg

Code for the paper "Pose2Seg: Detection Free Human Instance Segmentation" @ CVPR2019.
http://www.liruilong.cn/projects/pose2seg/index.html
MIT License
532 stars 136 forks source link

RuntimeError: CUDA out of memory in training #32

Open Mabozaid0X1 opened 4 years ago

Mabozaid0X1 commented 4 years ago

I follow all steps but when I run python train.py I find this error RuntimeError: CUDA out of memory. Tried to allocate 128.00 MiB (GPU 0; 4.00 GiB total capacity; 2.57 GiB already allocated; 74.77 MiB free; 2.85 GiB reserved in total by PyTorch) (malloc at ..\c10\cuda\CUDACachingAllocator.cpp:289) (no backtrace available)

KainingYing commented 3 years ago

maybe you should reduce the batchsize or use a bigger gpu memory machine