Skip to content

GitLab

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

Reduce time for docs test in CI (#239)

parent a8d93c54
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/nnp_training_force.py
View file @ 0ce36d82
...@@ -205,7 +205,7 @@ if 'scheduler' in checkpoint: ...@@ -205,7 +205,7 @@ if 'scheduler' in checkpoint:
############################################################################### ###############################################################################
# In the training loop, we need to compute force, and loss for forces # In the training loop, we need to compute force, and loss for forces
print("training starting from epoch", scheduler.last_epoch + 1) print("training starting from epoch", scheduler.last_epoch + 1)
max_epochs = 200 max_epochs = 20
early_stopping_learning_rate = 1.0E-5 early_stopping_learning_rate = 1.0E-5
force_coefficient = 1 # controls the importance of energy loss vs force loss force_coefficient = 1 # controls the importance of energy loss vs force loss
best_model_checkpoint = 'force-training-best.pt' best_model_checkpoint = 'force-training-best.pt'
......
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