Commit 3de0ea5b authored by sjromuel's avatar sjromuel
Browse files

d

parent 7e0a8872
......@@ -135,8 +135,8 @@ def train_unet(model, inputs, gt, weights, optimizer, filter_multiplier):
def init_weights(manual_seed):
### initializing weights ###
initializer = tf.initializers.glorot_uniform(seed=manual_seed)
#initializer = tf.keras.initializers.TruncatedNormal(seed=manual_seed)
#initializer = tf.initializers.glorot_uniform(seed=manual_seed)
initializer = tf.keras.initializers.TruncatedNormal(seed=manual_seed)
shapes = [ # filter_height, filter_width, in_channels, out_channels
# for conv2d_transpose: filter_height, filter_width, out_channels, in_channels
[3, 3, 1, 16],
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment