diff --git a/modules/deeplearning/srcnn.py b/modules/deeplearning/srcnn.py index 7ee52b335c25cc444c011df44168b498fb5e78b9..0af63accf0a776ffb7af78297e0e15e9e03af855 100644 --- a/modules/deeplearning/srcnn.py +++ b/modules/deeplearning/srcnn.py @@ -344,8 +344,7 @@ class SRCNN: activation = tf.nn.relu momentum = 0.99 - num_filters = 256 - num_res_blocks = 4 + num_filters = 64 input_2d = self.inputs[0] print('input: ', input_2d.shape)