search

snwagh / falcon-public

Implementation of protocols in Falcon
90 stars 46 forks source link

How to change the number of epochs when training networks #44

Closed khoaguin closed 2 years ago

khoaguin commented 2 years ago

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 commented 2 years ago

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

khoaguin commented 2 years ago

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

khoaguin commented 2 years ago

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?

Screenshot from 2022-07-05 10-12-37

snwagh commented 2 years ago

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.