-
Notifications
You must be signed in to change notification settings - Fork 3
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
added uncertainty estimation on dump interval for jaxmd #312
Conversation
also closes #309 |
for more information, see https://pre-commit.ci
pre-commit.ci autofix |
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
for more information, see https://pre-commit.ci
apax/md/simulate.py
Outdated
def body_fn(i, state): | ||
state, neighbor = state | ||
state, outer_step, neighbor = state # neighbors |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
don't understand the comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The comment is superfluous and can be removed.
apax/md/simulate.py
Outdated
if isinstance(state, simulate.NPTNoseHooverState): | ||
box = state.box | ||
apply_fn_kwargs = {} | ||
else: | ||
box = system.box | ||
apply_fn_kwargs = {"box": box} | ||
apply_fn_kwargs = {"box": box} # this might cause a bug |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
should we discuss this bevor merging?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The comment was just a reminder for myself to look into this. it's all working as intended. I'll remove the comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
After removing the comments it looks fine from me.
added temperature schedules
A full model is evaluated when a configuration is dumped.
Before we would evaluate an energy model at every time step but only collect the energy aver N steps, which was a bit redundant. Now we can save all model outputs.
closes #298