From b83b35a604f1b4d0d48162c9c8f82a2091ae0f05 Mon Sep 17 00:00:00 2001
From: Engin Eren <engin.eren@desy.de>
Date: Wed, 21 Sep 2022 15:27:42 +0200
Subject: [PATCH] comet error
---
wgan_ECAL_HCAL_3crit.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/wgan_ECAL_HCAL_3crit.py b/wgan_ECAL_HCAL_3crit.py
index 62bc90a..61cf522 100644
--- a/wgan_ECAL_HCAL_3crit.py
+++ b/wgan_ECAL_HCAL_3crit.py
@@ -365,7 +365,7 @@ def train(args, aDE, aDH, aD_H_E, aGE, aGH, device, train_loader, optimizer_d_E,
niter = epoch * len(train_loader) + batch_idx
experiment.log_metric("L_Gen_H_E", g_cost, step=niter)
- experiment.log_metric("L_aux_Esum", auxLoss, step=niter)
+ experiment.log_metric("L_aux_Esum", torch.mean(auxLoss), step=niter)
--
GitLab