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

minor

parent b0fc613c
Branches
No related tags found
No related merge requests found
......@@ -216,12 +216,12 @@ class UNET:
self.n_chans = 10
if TRIPLET:
self.n_chans *= 3
# self.X_img = tf.keras.Input(shape=(None, None, self.n_chans))
self.X_img = tf.keras.Input(shape=(30, 30, self.n_chans))
self.X_img = tf.keras.Input(shape=(None, None, self.n_chans))
# self.X_img = tf.keras.Input(shape=(30, 30, self.n_chans))
self.inputs.append(self.X_img)
# self.inputs.append(tf.keras.Input(shape=(None, None, self.n_chans)))
self.inputs.append(tf.keras.Input(shape=(30, 30, self.n_chans)))
self.inputs.append(tf.keras.Input(shape=(None, None, self.n_chans)))
# self.inputs.append(tf.keras.Input(shape=(30, 30, self.n_chans)))
self.DISK_CACHE = False
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment