limited number of epochs
agni47 opened this issue · 2 comments
agni47 commented
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
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
It works after setting visualize to 0. Thanks.