Update _potential.py - hard-code patience?

materialsvirtuallab / m3gnet

Materials graph network with 3-body interactions featuring a DFT surrogate crystal relaxer and a state-of-the-art property predictor.

BSD 3-Clause "New" or "Revised" License

238 stars 61 forks source link

Update _potential.py - hard-code patience? #22

Closed VvVLzy closed 2 years ago

VvVLzy commented 2 years ago

It seems that the patience=200 is hard-coded here? Making patience=early_stop_patience will also be helpful for continued training (monitor training progress across different runs).