This commit is contained in:
Nataniel Ruiz
2019-12-23 16:47:01 -04:00
parent ed0749d0ad
commit 41a2b97802
+2
View File
@@ -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