mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Print when wrapping

Browse Source
This commit is contained in:
James Betker 2020-08-25 18:08:46 -06:00
parent bae18c05e6
commit 3f60281da7
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/models/networks.py
View File

@ -143,6 +143,7 @@ def define_G(opt, net_key='network_G', scale=None):
class GradDiscWrapper(torch.nn.Module): class GradDiscWrapper(torch.nn.Module):
def __init__(self, m): def __init__(self, m):
super(GradDiscWrapper, self).__init__() super(GradDiscWrapper, self).__init__()
print("Wrapping a discriminator..")
self.m = m self.m = m
def forward(self, x, lr): def forward(self, x, lr):