How to parameterize number of epochs

[Tapojit Debnath]

Hi,

I cannot find a way to limit the number of epochs during CNN training. The only way I can stop training right now is by killing the process. Is there a way to add a limited number of epochs as an argument? 

[Harold]

Hello,

Thanks for your interest and question.

If you're using the experiment framework, then `epoch-count` is a parameter to the `train-n` function:

https://github.com/thinktopic/cortex/blob/master/experiment/src/cortex/experiment/train.clj#L171

Here are some examples of such use:

https://github.com/thinktopic/cortex/blob/master/experiment/test/cortex/experiment/logistic_test.clj#L42

https://github.com/thinktopic/cortex/blob/master/examples/mnist-classification/src/mnist_classification/core.clj#L205

If you're calling `execute/train` directly, you'll need to manage quitting after a number of epochs yourself, here is how the experiment framework does it:

https://github.com/thinktopic/cortex/blob/master/experiment/src/cortex/experiment/train.clj#L124

Hope that helps, let us know if there's more info we can provide,

-Harold

[Tapojit Debnath]

It worked! Thanks a lot for pointing that out!

[Harold]

Glad to hear it. You are welcome.

-Harold