next
This commit is contained in:
@@ -66,8 +66,10 @@ class Generator(nn.Module):
|
||||
|
||||
# Get intermediate feature maps
|
||||
for layer in self.main:
|
||||
print('test')
|
||||
x = layer(x)
|
||||
feature_maps.append(x)
|
||||
print(feature_maps.shape)
|
||||
|
||||
return x, feature_maps
|
||||
|
||||
|
||||
Reference in New Issue
Block a user