added fallback local init

2026-09-11 12:31:23 +08:00 · 2019-07-13 22:16:10 -04:00
parent eb05fa316f
commit a2639c6894
1 changed files with 1 additions and 1 deletions
@@ -599,7 +599,7 @@ class Trainer(TrainerIO):
        try:
            loss = output['loss']
        except Exception as e:
-            if type(loss) is torch.Tensor:
+            if type(output) is torch.Tensor:
                loss = output

        self.__add_tqdm_metrics(model_specific_tqdm_metrics_dic)