fix combined loss
@@ -53,7 +53,8 @@ class CombinedLoss(nn.Layer):
for key in loss:
if key == "loss":
loss_all += loss[key] * weight
- # else:
+ else:
+ loss["{}_{}".format(key, idx)] = loss[key]
# loss[f"{key}_{idx}"] = loss[key]
loss_dict.update(loss)
loss_dict["loss"] = loss_all
