Avoid running validation set twice

materialsvirtuallab / megnet

Graph Networks as a Universal Machine Learning Framework for Molecules and Crystals

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

497 stars 155 forks source link

Avoid running validation set twice #365

Closed dgaines2 closed 2 years ago

dgaines2 commented 2 years ago

If save_checkpoint, set val_generator and steps_per_val to None. This avoids issues with accidentally running validation twice -- matching the original behavior before PR #364