From 41a2b978020b6048c6b511c7058f3a9852957d35 Mon Sep 17 00:00:00 2001
From: Nataniel Ruiz <nruiz@Nataniels-MacBook-Pro.local>
Date: Mon, 23 Dec 2019 16:47:01 -0400
Subject: [PATCH] next

---
 stargan/model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/stargan/model.py b/stargan/model.py
index 82f476d..79d2f73 100644
--- a/stargan/model.py
+++ b/stargan/model.py
@@ -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