diff options
| author | aktersnurra <gustaf.rydholm@gmail.com> | 2021-01-07 20:10:54 +0100 |
|---|---|---|
| committer | aktersnurra <gustaf.rydholm@gmail.com> | 2021-01-07 20:10:54 +0100 |
| commit | ff9a21d333f11a42e67c1963ed67de9c0fda87c9 (patch) | |
| tree | afee959135416fe92cf6df377e84fb0a9e9714a0 /src/training/trainer | |
| parent | 25b5d6983d51e0e791b96a76beb7e49f392cd9a8 (diff) | |
Minor updates.
Diffstat (limited to 'src/training/trainer')
| -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("_") |