Small bug

[Pankaj Daga]

I think you have a small bug in your code, you need to reset the early_stop variable as:

        self.best_score = score
        self.save_checkpoint(val_loss, model)
        self.counter = 0
        self.early_stop = False  # reset here

Otherwise, if you check this flag from outside, it will possibly stop earlier depending on what happened in the previous iterations.

[Pankaj Daga]

Sorry my bad.