materialsvirtuallab/megnet

Early stopping only works when save_checkpoint set True

Nokimann opened this issue · 1 comments

Nokimann commented

The early stoping works if

model.train(save_checkpoint=True, patience=1)

The early stopping doesn't work if

model.train(save_checkpoint=False, patience=1)

chc273 commented

You can use keras early stop callback

  • 👍1