Skip to content
Snippets Groups Projects
Commit 54f24eb1 authored by gong1's avatar gong1
Browse files

fix import model_modules.video_prediction issues

parent ba058969
No related branches found
No related tags found
No related merge requests found
Pipeline #55958 failed
...@@ -1065,7 +1065,7 @@ def get_norm_layer(layer_type): ...@@ -1065,7 +1065,7 @@ def get_norm_layer(layer_type):
elif layer_type == 'layer': elif layer_type == 'layer':
layer = tf.contrib.layers.layer_norm layer = tf.contrib.layers.layer_norm
elif layer_type == 'instance': 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 layer = fused_instance_norm
elif layer_type == 'none': elif layer_type == 'none':
layer = tf.identity layer = tf.identity
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment