itayhubara/BinaryNet

limited number of epochs

agni47 opened this issue · 2 comments

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.

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

It works after setting visualize to 0. Thanks.