From ff9a21d333f11a42e67c1963ed67de9c0fda87c9 Mon Sep 17 00:00:00 2001 From: aktersnurra Date: Thu, 7 Jan 2021 20:10:54 +0100 Subject: Minor updates. --- src/training/trainer/callbacks/wandb_callbacks.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'src/training/trainer/callbacks') 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("_") -- cgit v1.2.3-70-g09d2