Skip to content

Commit cb2d9b0

Browse files
committed
Update logging name conventation in line with other updates
1 parent e48e634 commit cb2d9b0

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

src/model/unet_interpolated.py

+4-7
Original file line numberDiff line numberDiff line change
@@ -76,14 +76,11 @@ def test_epoch_end(self, outputs):
7676
tensorboard_logs[f"test_loss_{n}"] = torch.stack(
7777
[d[str(n)] for d in [x["log"]["test_loss"] for x in outputs if x]]
7878
).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-
]
79+
tensorboard_logs[f"acc_{n}"] = torch.stack(
80+
[d[str(n)] for d in [x["log"]["acc_test"] for x in outputs if x]]
8481
).mean()
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]]
82+
tensorboard_logs[f"mae_{n}"] = torch.stack(
83+
[d[str(n)] for d in [x["log"]["mae_test"] for x in outputs if x]]
8784
).mean()
8885

8986
return {

0 commit comments

Comments
 (0)