nyud_layer: how to reduce data size

6 views
Skip to first unread message

gio

unread,
Sep 12, 2017, 3:57:22 PM9/12/17
to Caffe Users
Hi all, I've failed to train fcn32s on nyud dataset because I don't have enough memory on my gpu, and so the training end with "cuda out of memory error". I want to reduce the size of the image, depth, hha data. Does anyone know how to modify the nyud_layer for the loading of the dataset, in order to make this resize step? 
Reply all
Reply to author
Forward
0 new messages