bug fix for loss error message

2022-12-27 21:38:56 +08:00
parent 0915e989ca
commit bf8d03658e
1 changed files with 1 additions and 1 deletions
@@ -65,7 +65,7 @@ class LossAggregator():
            else:
                if isinstance(v, dict):
                    raise ValueError(
-                        "The key %s in -Trainng-Feat- should be stated as the log_prefix of a certain loss defined in your loss_cfg."%v
+                        "The key %s in -Trainng-Feat- should be stated in your loss_cfg as log_prefix."%k
                    )
                elif is_tensor(v):
                    _ = v.mean()