How to change the number of epochs when training networks

[khoaguin]

Hi,

How can I define the number of training epochs for different networks? I have found that there is a variable named NUM_ITERATIONS in main.cpp, but not sure what it is, and also we cannot change it in makefile either.

Thank you.

[snwagh]

The variable is set src/globals.h so you can change it there.

[khoaguin]

I tried to change the variable NO_OF_EPOCHS in src/globals.h to different values, but nothing really happened

[khoaguin]

From the train function, I think that the NUM_ITERATIONS variable is actually the number of epochs to train the neural network. Am I wrong?

[snwagh]

So the codebase is designed to use NUM_ITERATIONS only, if you want to set using the number of epochs, you comment out this line and uncomment this line.