Print when wrapping

2020-08-25 18:08:46 -06:00 · 2020-08-25 18:08:46 -06:00 · 3f60281da7
commit 3f60281da7
parent bae18c05e6
1 changed files with 1 additions and 0 deletions
--- 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):