diff --git a/modules/deeplearning/espcn.py b/modules/deeplearning/espcn.py index f8d30d5be0842eb8209dafe06bd19e7e1981554f..a7a9395c333364b40d14b9b92e4ee2d1b6eb8687 100644 --- a/modules/deeplearning/espcn.py +++ b/modules/deeplearning/espcn.py @@ -437,11 +437,9 @@ class ESPCN: print(conv.shape) conv = tf.keras.layers.Conv2DTranspose(num_filters // 4, kernel_size=1, strides=1, padding=padding, activation=activation)(conv) - conv.trainable = False print(conv.shape) conv = tf.keras.layers.Conv2DTranspose(num_filters // 4, kernel_size=1, strides=1, padding=padding, activation=activation)(conv) - conv.trainable = False print(conv.shape) #self.logits = tf.keras.layers.Conv2D(1, kernel_size=1, strides=1, padding=padding, name='probability', activation=tf.nn.sigmoid)(conv)