Update checkpoint_path

Nested/trainers/BaseTrainer.py

@@ -107,7 +107,8 @@ class BaseTrainer:
         :param checkpoint_path: str - path/to/checkpoints
         :return: None
         """
-        checkpoint_path = natsort.natsorted("checkpoint_2.pt")
+        # checkpoint_path = natsort.natsorted(glob.glob(f"{checkpoint_path}/checkpoint_*.pt"))
+        checkpoint_path = natsort.natsorted(checkpoint_path)
         # checkpoint_path = checkpoint_path[-1]
 
         logger.info("Loading checkpoint %s", checkpoint_path)

Nested/utils/helpers.py

@@ -97,7 +97,7 @@ def load_checkpoint(model_path):
     train_config.trainer_config["kwargs"]["loss"] = loss
 
     tagger = load_object(train_config.trainer_config["fn"], train_config.trainer_config["kwargs"])
-    checkpoint_path = hf_hub_download(repo_id="SinaLab/Nested", filename="checkpoints/checkpoint_2.pt")
+    checkpoint_path = hf_hub_download(repo_id="SinaLab/Nested", filename="checkpoints")
 
     tagger.load(checkpoint_path)
     return tagger, tag_vocab, train_config

app.py

@@ -27,6 +27,7 @@ checkpoint_path = hf_hub_download(
     repo_id="SinaLab/Nested",
     filename="checkpoints/checkpoint_2.pt"
 )
+print("checkpoint_path : ", checkpoint_path)
 
 args_path = hf_hub_download(
     repo_id="SinaLab/Nested",