diff options
Diffstat (limited to 'src/training')
| -rw-r--r-- | src/training/prepare_experiments.py | 2 | ||||
| -rw-r--r-- | src/training/run_experiment.py | 2 | ||||
| -rw-r--r-- | src/training/trainer/callbacks/wandb_callbacks.py | 4 |
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("_") |