From 54f24eb1d21b5075875a8834c7a7d91be96b0a0f Mon Sep 17 00:00:00 2001
From: gong1 <b.gong@fz-juelich.de>
Date: Mon, 21 Dec 2020 19:58:36 +0100
Subject: [PATCH] fix import model_modules.video_prediction issues
---
video_prediction_tools/model_modules/video_prediction/ops.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/video_prediction_tools/model_modules/video_prediction/ops.py b/video_prediction_tools/model_modules/video_prediction/ops.py
index 1d2e9f2e..7e41af95 100644
--- a/video_prediction_tools/model_modules/video_prediction/ops.py
+++ b/video_prediction_tools/model_modules/video_prediction/ops.py
@@ -1065,7 +1065,7 @@ def get_norm_layer(layer_type):
elif layer_type == 'layer':
layer = tf.contrib.layers.layer_norm
elif layer_type == 'instance':
- from video_prediction.layers import fused_instance_norm
+ from model_modules.video_prediction.layers import fused_instance_norm
layer = fused_instance_norm
elif layer_type == 'none':
layer = tf.identity
--
GitLab