From 309ff2b6fa54aab06427df218ce44dfd351a2532 Mon Sep 17 00:00:00 2001 From: mir-of Date: Mon, 1 Jun 2020 23:01:51 +0800 Subject: [PATCH] minor fix --- cnn_e2e/resnet_model.py | 1 + 1 file changed, 1 insertion(+) diff --git a/cnn_e2e/resnet_model.py b/cnn_e2e/resnet_model.py index 01e91b1..c8d9f94 100755 --- a/cnn_e2e/resnet_model.py +++ b/cnn_e2e/resnet_model.py @@ -22,6 +22,7 @@ def _conv2d( weight_initializer=flow.variance_scaling_initializer( 2, 'fan_in', 'random_normal', data_format="NCHW"), weight_regularizer=flow.regularizers.l2(1.0/32768), + trainable=True, ): weight = flow.get_variable( name + "-weight", -- GitLab