From 32e91a688f670358ba6913fa9d6a6f3fe136d844 Mon Sep 17 00:00:00 2001 From: PatWie Date: Mon, 5 Nov 2018 13:21:56 +0100 Subject: [PATCH] fix new keras interface: allows inference during training with same weights --- layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/layers.py b/layers.py index 8ee136f..93fc582 100644 --- a/layers.py +++ b/layers.py @@ -24,7 +24,7 @@ from user_ops import flex_convolution_transpose as _flex_convolution_transpose from tensorflow.python.keras import activations from tensorflow.python.keras import initializers -from tensorflow.python.keras.engine.base_layer import Layer +from tensorflow.python.layers.base import Layer from tensorflow.python.util.tf_export import tf_export from tensorflow.python.framework import tensor_shape from tensorflow.python.framework import ops