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: ...@@ -216,12 +216,12 @@ class UNET:
self.n_chans = 10 self.n_chans = 10
if TRIPLET: if TRIPLET:
self.n_chans *= 3 self.n_chans *= 3
# self.X_img = tf.keras.Input(shape=(None, None, 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.X_img = tf.keras.Input(shape=(30, 30, self.n_chans))
self.inputs.append(self.X_img) 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=(None, None, self.n_chans)))
self.inputs.append(tf.keras.Input(shape=(30, 30, self.n_chans))) # self.inputs.append(tf.keras.Input(shape=(30, 30, self.n_chans)))
self.DISK_CACHE = False self.DISK_CACHE = False
... ...
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment