From 762498e24165002c9aebfa2a663ead18c9a2a904 Mon Sep 17 00:00:00 2001 From: Varun Prakash <98808045+VarunPrakashEX2@users.noreply.github.com> Date: Thu, 7 Apr 2022 10:10:19 +0530 Subject: [PATCH] Fixed the no. of nodes in the first conv2d layer. Fix the number of nodes in the first conv2d layer as the original published paper. --- model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model.py b/model.py index 57d61f05..7f66136f 100755 --- a/model.py +++ b/model.py @@ -11,7 +11,7 @@ def get_G(input_shape): g_init = tf.random_normal_initializer(1., 0.02) nin = Input(input_shape) - n = Conv2d(64, (3, 3), (1, 1), act=tf.nn.relu, padding='SAME', W_init=w_init)(nin) + n = Conv2d(64, (9, 9), (1, 1), act=tf.nn.relu, padding='SAME', W_init=w_init)(nin) temp = n # B residual blocks