Train.py use early stoping method

I use this web(https://github.com/dusty-nv/pytorch-classification/blob/5107de352cade326e7aacd44bf8d625b7055fd4e/train.py)to use train.py. I want to add the early stopping method to train.py. How to do it?

Hi,

You can terminate the process once it reaches the expected accuracy directly.
Then use the --resume flag if you want to resume the training job again.

Thanks.

I am very sorry to say. Would it be convenient for you to show or write the early_stopping code with train.py to me?

Or how to use dropout method on train.py?

Hi,

Dropout should be added into the model directly.
Here are some related discussion for your reference:

Thanks.

I use this URL (https://github.com/dusty-nv/jetson-inference/blob/master/docs/pytorch-collect.md) to go to the training step. How can I change the weight decay of train.py (https://github.com/dusty-nv/pytorch-classification/blob/5107de352cade326e7aacd44bf8d625b7055fd4e/train.py) without affecting the bias value?

Hi,

The training weight is updated via back propagation with the loss value automatically.
If you want to freeze the bias value, please try to implement a custom updater or set the trainable tag into OFF.

Thanks.