Debugging, set up load_ckpt for eval

smsharma · Jan 17, 2024 · fd46868 · fd46868
1 parent 8738a0f
commit fd46868
Show file tree

Hide file tree

Showing 5 changed files with 326 additions and 70 deletions.
diff --git a/configs/base.py b/configs/base.py
@@ -50,6 +50,9 @@ def get_config():
     training.ckpt_best_metric = "loss"  # "loss" or "top_xx_accuracy"
     training.ckpt_best_metric_best_mode = "min"  # "max" or "min" 
     training.ckpt_keep_top_n = 3  # Save the top `ckpt_keep_top_n` checkpoints based on `ckpt_best_metric`
+    training.load_ckpt = True
+    training.ckpt_run_name = "ancient-pine-88"
+
 
     # Sum1 options
     config.sum1 = sum1 = ml_collections.ConfigDict()

diff --git a/notebooks/04_retrieval_acc.ipynb b/notebooks/04_retrieval_acc.ipynb