We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e48e634 commit cb2d9b0Copy full SHA for cb2d9b0
src/model/unet_interpolated.py
@@ -76,14 +76,11 @@ def test_epoch_end(self, outputs):
76
tensorboard_logs[f"test_loss_{n}"] = torch.stack(
77
[d[str(n)] for d in [x["log"]["test_loss"] for x in outputs if x]]
78
).mean()
79
- tensorboard_logs[f"test_acc_{n}"] = torch.stack(
80
- [
81
- d[str(n)]
82
- for d in [x["log"]["n_correct_pred_test"] for x in outputs if x]
83
- ]
+ tensorboard_logs[f"acc_{n}"] = torch.stack(
+ [d[str(n)] for d in [x["log"]["acc_test"] for x in outputs if x]]
84
85
- tensorboard_logs[f"abs_error_{n}"] = torch.stack(
86
- [d[str(n)] for d in [x["log"]["abs_error_test"] for x in outputs if x]]
+ tensorboard_logs[f"mae_{n}"] = torch.stack(
+ [d[str(n)] for d in [x["log"]["mae_test"] for x in outputs if x]]
87
88
89
return {
0 commit comments