Skip to content

[Feature Request] Update the energy bias after the final step of training #3408

@njzjz

Description

@njzjz

Summary

Add an option to update the energy bias after the final step of training.

Detailed Description

Sometimes, the predicted energy is shifted compared to the reference energy, such as #2935. I think it might be because the energy loss is too small (the force loss cannot train the energy bias). However, we have an easy way to improve it: do update_energy_bias after the final step of the training. It just changes the energy bias and will not break other things. We already have the code for update_energy_bias, so we can add an option to call it.
(in theory, it is equivalent to --finetune with 0 steps, but a user won't come up with it.)

Further Information, Files, and Links

No response

Metadata

Metadata

Assignees

Projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions