Minor updates.

3 files changed, 5 insertions, 3 deletions

diff --git a/src/training/prepare_experiments.py b/src/training/prepare_experiments.py
index 6e20bcd..21997af 100644
--- a/src/training/prepare_experiments.py
+++ b/src/training/prepare_experiments.py
@@ -14,7 +14,7 @@ def run_experiments(experiments_filename: str) -> None:
     for index in range(num_experiments):
         experiment_config = experiments_config["experiments"][index]
         experiment_config["experiment_group"] = experiments_config["experiment_group"]
-        cmd = f"python training/run_experiment.py --gpu=-1 --save '{json.dumps(experiment_config)}'"
+        cmd = f"poetry run run-experiment --gpu=-1 --save '{json.dumps(experiment_config)}'"
         print(cmd)
 
 
diff --git a/src/training/run_experiment.py b/src/training/run_experiment.py
index 0167725..2c9a196 100644
--- a/src/training/run_experiment.py
+++ b/src/training/run_experiment.py
@@ -256,7 +256,7 @@ def run_experiment(
     # Load from checkpoint if resuming an experiment.
     resume = False
     if checkpoint is not None or pretrained_weights is not None:
-        resume = True
+        # resume = True
         _load_from_checkpoint(model, model_dir, pretrained_weights)
 
     logger.info(f"The class mapping is {model.mapping}")
diff --git a/src/training/trainer/callbacks/wandb_callbacks.py b/src/training/trainer/callbacks/wandb_callbacks.py
index df1fd8f..20414df 100644
--- a/src/training/trainer/callbacks/wandb_callbacks.py
+++ b/src/training/trainer/callbacks/wandb_callbacks.py
@@ -110,7 +110,9 @@ class WandbImageLogger(Callback):
             if isinstance(self.targets[i], list):
                 ground_truth = "".join(
                     [
-                        self.model.mapper(int(target_index))
+                        self.model.mapper(int(target_index) - 26)
+                        if target_index > 35
+                        else self.model.mapper(int(target_index))
                         for target_index in self.targets[i]
                     ]
                 ).rstrip("_")