Mentions légales du service

Skip to content
Snippets Groups Projects
Commit 54bc1946 authored by PaulWawerek-L's avatar PaulWawerek-L
Browse files

print best loss from checkpoint

parent 86ec4139
No related branches found
No related tags found
1 merge request!2OSLO-IC
Hide whitespace changes
Inline Side-by-side
main_sphere_compression.py
+ 1
− 0
View file @ 54bc1946
...@@ -536,6 +536,7 @@ def main(): ...@@ -536,6 +536,7 @@ def main():
if "best_loss" in checkpoint: if "best_loss" in checkpoint:
best_loss = checkpoint["best_loss"] best_loss = checkpoint["best_loss"]
print("best_loss loaded=", best_loss)
if not args.no_scheduler and not args.test_data: if not args.no_scheduler and not args.test_data:
if "scheduler_state_dict" in checkpoint: if "scheduler_state_dict" in checkpoint:
......
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