How can I reduce the number of Epochs for training?

lindawangg / COVID-Net

COVID-Net Open Source Initiative

Other

1.15k stars 480 forks source link

How can I reduce the number of Epochs for training? #68

Closed jchilela closed 3 years ago

jchilela commented 4 years ago

I am training in a not so powerful computer, I would like to reduce the number of epochs to 3 or 5 only, so I can have a trained model by me. For now I Know it has 22 epochs by 1678 each it is taking long.

Where can I change this

haydengunraj commented 4 years ago

When running train_tf.py, you can set the number of epochs using the --epochs argument.