diff --git a/wgan_ECAL_HCAL_3crit.py b/wgan_ECAL_HCAL_3crit.py
index 62bc90a4511574e75f105fa0b87f605e51dc5c4d..61cf522c34cd9d36038d9e93c8ee66d010ee3e9c 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)