Skip to content
Snippets Groups Projects
Commit 7eb974e2 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI
Browse files

[simplecnn] fix a bug with slim architecture

parent e9c6d71b
Branches
Tags
1 merge request!72[predict_bio] Some fixes and new features
......@@ -271,7 +271,9 @@ def slim_architecture(
weights_regularizer=slim.l2_regularizer(weight_decay),
normalizer_fn=slim.batch_norm,
normalizer_params=batch_norm_params):
with tf.variable_scope('SimpleCNN', reuse=reuse):
with tf.variable_scope('SimpleCNN', reuse=reuse), slim.arg_scope(
[slim.batch_norm, slim.dropout],
is_training=mode == tf.estimator.ModeKeys.TRAIN):
initializer = tf.contrib.layers.xavier_initializer()
name = 'conv1'
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment