diff --git a/codes/models/networks.py b/codes/models/networks.py
index 19387325..f29d5450 100644
--- a/codes/models/networks.py
+++ b/codes/models/networks.py
@@ -143,6 +143,7 @@ def define_G(opt, net_key='network_G', scale=None):
 class GradDiscWrapper(torch.nn.Module):
     def __init__(self, m):
         super(GradDiscWrapper, self).__init__()
+        print("Wrapping a discriminator..")
         self.m = m
 
     def forward(self, x, lr):