search

itayhubara / BinaryNet

Training Deep Neural Networks with Weights and Activations Constrained to +1 or -1
296 stars 95 forks source link

limited number of epochs #6

Closed agni47 closed 8 years ago

agni47 commented 8 years ago

Main_BinaryNet_MNIST.lua runs only for 3 epochs, regardless of the number of epochs i specify in line 32 cmd:option('-epoch', -1, 'number of epochs to train, -1 for unbounded')

everytime it achieves a test accuracy of 94-95% and stops.

Itay-Hubara commented 8 years ago

For some reason you have a problem plotting the convergence graph (Main_BinaryNet_MNIST.lua - line 324). Try setting opt.visualize to 0.

Best, Itay

agni47 commented 8 years ago

It works after setting visualize to 0. Thanks.