search

ishay2b / VanillaCNN

Implementation of the Vanilla CNN described in the paper: Yue Wu and Tal Hassner, "Facial Landmark Detection with Tweaked Convolutional Neural Networks", arXiv preprint arXiv:1511.04031, 12 Nov. 2015. See project page for more information about this project. http://www.openu.ac.il/home/hassner/projects/tcnn_landmarks/ Written by Ishay Tubi : ishay2b [at] gmail [dot] com https://www.l
188 stars 80 forks source link

train error，Creating layer loss #14

Closed janicevidal closed 6 years ago

janicevidal commented 6 years ago

MemoryError _20180315_111313 why？seeking for help

ishay2b commented 6 years ago

Hi, Looks like a memory issue. The batch size is too big for your computer, try reduce batch size. If you run out on gpu, try using cpu instead, I trained it using cpu with 8G ram.