Merge pull request #173 from Saren-Arterius/master

Fix requirements
tensorlayer · Nov 3, 2019 · e7d26c4 · e7d26c4
2 parents 5734a98 + 7ef4d32
commit e7d26c4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/model.py b/model.py
@@ -24,7 +24,7 @@ def get_G(input_shape):
         n = nn
 
     n = Conv2d(64, (3, 3), (1, 1), padding='SAME', W_init=w_init, b_init=None)(n)
-    n = BatchNorm(gamma_init=g_init)(n)
+    n = BatchNorm2d(gamma_init=g_init)(n)
     n = Elementwise(tf.add)([n, temp])
     # B residual blacks end
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,2 +1,4 @@
 tensorlayer>=2.0.0
-tensorflow>=2.0.0
+tensorflow>=2.0.0b1
+numpy==1.16.1
+easydict==1.9