Skip to content

GitLab

Unverified Commit f93a9ba4 authored by Gao, Xiang's avatar Gao, Xiang Committed by GitHub
Browse files

Rename to better describe what it is (#58)

parent 5bb66915
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/nnp_training.py
View file @ f93a9ba4
...@@ -129,7 +129,7 @@ def log_time(trainer): ...@@ -129,7 +129,7 @@ def log_time(trainer):
def log_loss_and_time(trainer): def log_loss_and_time(trainer):
iteration = trainer.state.iteration iteration = trainer.state.iteration
rmse = hartree2kcal(math.sqrt(trainer.state.output)) rmse = hartree2kcal(math.sqrt(trainer.state.output))
writer.add_scalar('training_rmse_vs_iteration', rmse, iteration) writer.add_scalar('training_atomic_rmse_vs_iteration', rmse, iteration)
trainer.run(training, max_epochs=parser.max_epochs) trainer.run(training, max_epochs=parser.max_epochs)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment