From 3f60281da70cfa4b7525f3b7aef23527ebf3d0df Mon Sep 17 00:00:00 2001 From: James Betker Date: Tue, 25 Aug 2020 18:08:46 -0600 Subject: [PATCH] Print when wrapping --- codes/models/networks.py | 1 + 1 file changed, 1 insertion(+) 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):