Как сделать раннюю остановку в классификации изображений pytorch

Question

Я новичок в Pytorch и машинном обучении. Я следую этому уроку в этом уроке https://www.learnopencv.com/image-classification-using-transfer-learning-in-pytorch/ и использую свой собственный набор данных. Тогда у меня возникла та же проблема в этом уроке, но я не знаю, как сделать раннюю остановку в pytorch, и, если вам лучше, не создавайте процесс ранней остановки, пожалуйста, сообщите мне.

Answer 1

Попробуйте с приведенным ниже кодом.

     # Check early stopping condition
     if epochs_no_improve == n_epochs_stop:
        print('Early stopping!' )
        early_stop = True
        break
     else:
        continue
     break
if early_stop:
    print("Stopped")
    break