search

raghudeep / netwarp_public

Semantic Video CNNs through Representation Warping. ICCV 2017
BSD 3-Clause "New" or "Revised" License
69 stars 20 forks source link

required GPU memory #4

Open amiltonwong opened 7 years ago

amiltonwong commented 7 years ago

Hi, @raghudeep,

How much GPU memory is required to run the segmentation? I got "out of memory" error using my Titan-Black 6GB GPU card. BTW, could I run the segmentation by reducing the batch number or other settings?

Thanks!

akshata-bhat commented 6 years ago

Hi @raghudeep,

I am running into the following error:

Loaded models/pspnet101_cityscapes_conv5_4netwarp_deploy.prototxt with models/pspnet101_cityscapes_conv5_4netwarp.caffemodel for GPU 0

  1. lindau_000019_000000 3:19:21PM EDT on May 30, 2018 on GPU 0 /usr/lib/python2.7/dist-packages/numpy/core/fromnumeric.py:225: VisibleDeprecationWarning: converting an array with ndim > 0 to an index will result in an error in the future return reshape(newshape, order=order) F0530 15:19:22.213881 2557 syncedmem.cpp:71] Check failed: error == cudaSuccess (2 vs. 0) out of memory Check failure stack trace: Aborted (core dumped)

Can you help me with this?

Thanks