other errors and typos

.gitlab-ci.yml

@@ -21,4 +21,4 @@ check_python_run:
   image: pytorch/pytorch:1.9.0-cuda10.2-cudnn7-runtime
   script: 
     - pip install pyflakes
-    - pyflakes $CI_PROJECT_DIR/wganHCAL.py
+    - pyflakes $CI_PROJECT_DIR/wgan_ECAL_HCAL_2crit.py

wgan_ECAL_HCAL_2crit.py

@@ -176,7 +176,7 @@ def train(args, aDE, aDH, aGE, aGH, device, train_loader, optimizer_d_E, optimiz
         if batch_idx % args.log_interval == 0:
             print('Critic --> Train Epoch: {} [{}/{} ({:.0f}%)]\tloss={:.4f}'.format(
                 epoch, batch_idx * len(dataH), len(train_loader.dataset),
-                100. * batch_idx / len(train_loader), disc_cost.item()))
+                100. * batch_idx / len(train_loader), disc_cost_H_E.item()))
             niter = epoch * len(train_loader) + batch_idx
             experiment.log_metric("L_crit_E", disc_cost_E, step=niter)
             experiment.log_metric("L_crit_H_E", disc_cost_H_E, step=niter)