Fix problem with normalization layer lambda.

gmalivenko · Jun 13, 2019 · 83a8c58 · 83a8c58
1 parent 3fcf870
commit 83a8c58
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pytorch2keras/normalization_layers.py b/pytorch2keras/normalization_layers.py
@@ -100,6 +100,7 @@ def convert_instancenorm(params, w_name, scope_name, inputs, layers, weights, na
         beta = weights[bias_name].numpy()
 
     def target_layer(x, epsilon=params['epsilon'], gamma=gamma, beta=beta):
+        import tensorflow as tf
         layer = tf.contrib.layers.instance_norm(
             x,
             param_initializers={'beta': tf.constant_initializer(beta), 'gamma': tf.constant_initializer(gamma)},