global: debug inference

2024-11-01 15:47:11 +00:00
parent 985a08d89c
commit ecd4cfa806
4 changed files with 113 additions and 43 deletions
--- a/configs/server/server_train_config.yaml
+++ b/configs/server/server_train_config.yaml
@@ -9,7 +9,7 @@ runner:
  experiment:
    name: train_ab_global_only
    root_dir: "experiments"
-    use_checkpoint: False
+    use_checkpoint: True
    epoch: -1 # -1 stands for last epoch
    max_epochs: 5000
    save_checkpoint_interval: 1