Skip to content
Snippets Groups Projects
Commit ae4017fd authored by tomrink's avatar tomrink
Browse files

minor

parent 0e7c1bb1
Branches
No related tags found
No related merge requests found
......@@ -356,7 +356,7 @@ class SRCNN:
print('input: ', conv.shape)
# conv = conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, padding=padding)(input_2d)
conv = conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, kernel_initializer='he_uniform', activation=activation, padding='VALID')(input_2d)
conv = conv_b = tf.keras.layers.Conv2D(num_filters, kernel_size=3, kernel_initializer='he_uniform', activation=activation, padding='SAME')(input_2d)
print(conv.shape)
if NOISE_TRAINING:
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment