diff options
Diffstat (limited to 'src/training/trainer/callbacks/wandb_callbacks.py')
| -rw-r--r-- | src/training/trainer/callbacks/wandb_callbacks.py | 4 |
1 files changed, 3 insertions, 1 deletions
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("_") |