From 3e471303e5723f2ab3de42689bec38fd618ba4e1 Mon Sep 17 00:00:00 2001 From: Michael <m.langguth@fz-juelich.de> Date: Thu, 20 May 2021 17:40:16 +0200 Subject: [PATCH] Corrected inclusion of model_helpers in convLSTM, GAN etc. . --- .../model_modules/video_prediction/models/convLSTM_GAN_model.py | 2 +- .../model_modules/video_prediction/models/mcnet_model.py | 2 +- .../model_modules/video_prediction/models/vanilla_GAN_model.py | 2 +- .../video_prediction/models/vanilla_convLSTM_model.py | 2 +- .../model_modules/video_prediction/models/vanilla_vae_model.py | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/video_prediction_tools/model_modules/video_prediction/models/convLSTM_GAN_model.py b/video_prediction_tools/model_modules/video_prediction/models/convLSTM_GAN_model.py index 972f9557..092cf81a 100644 --- a/video_prediction_tools/model_modules/video_prediction/models/convLSTM_GAN_model.py +++ b/video_prediction_tools/model_modules/video_prediction/models/convLSTM_GAN_model.py @@ -2,7 +2,7 @@ __email__ = "b.gong@fz-juelich.de" __author__ = "Bing Gong,Yanji" __date__ = "2021-04-13" -from model_helpers import set_and_check_pred_frames +from model_modules.video_prediction.models.model_helpers import set_and_check_pred_frames import tensorflow as tf from model_modules.video_prediction.layers import layer_def as ld from model_modules.video_prediction.layers.BasicConvLSTMCell import BasicConvLSTMCell diff --git a/video_prediction_tools/model_modules/video_prediction/models/mcnet_model.py b/video_prediction_tools/model_modules/video_prediction/models/mcnet_model.py index 61fdb912..2c755417 100644 --- a/video_prediction_tools/model_modules/video_prediction/models/mcnet_model.py +++ b/video_prediction_tools/model_modules/video_prediction/models/mcnet_model.py @@ -7,7 +7,7 @@ import itertools import numpy as np import tensorflow as tf -from model_helpers import set_and_check_pred_frames +from model_modules.video_prediction.models.model_helpers import set_and_check_pred_frames from model_modules.video_prediction.models import BaseVideoPredictionModel from model_modules.video_prediction.ops import dense, pad2d, conv2d, flatten, tile_concat from model_modules.video_prediction.layers.BasicConvLSTMCell import BasicConvLSTMCell diff --git a/video_prediction_tools/model_modules/video_prediction/models/vanilla_GAN_model.py b/video_prediction_tools/model_modules/video_prediction/models/vanilla_GAN_model.py index 747e1933..112eaf31 100644 --- a/video_prediction_tools/model_modules/video_prediction/models/vanilla_GAN_model.py +++ b/video_prediction_tools/model_modules/video_prediction/models/vanilla_GAN_model.py @@ -13,7 +13,7 @@ This code implement take the following as references: import tensorflow as tf -from model_helpers import set_and_check_pred_frames +from model_modules.video_prediction.models.model_helpers import set_and_check_pred_frames from model_modules.video_prediction.layers import layer_def as ld from tensorflow.contrib.training import HParams diff --git a/video_prediction_tools/model_modules/video_prediction/models/vanilla_convLSTM_model.py b/video_prediction_tools/model_modules/video_prediction/models/vanilla_convLSTM_model.py index 1780b2e8..570ece3e 100644 --- a/video_prediction_tools/model_modules/video_prediction/models/vanilla_convLSTM_model.py +++ b/video_prediction_tools/model_modules/video_prediction/models/vanilla_convLSTM_model.py @@ -2,7 +2,7 @@ __email__ = "b.gong@fz-juelich.de" __author__ = "Bing Gong, Scarlet Stadtler,Michael Langguth" __date__ = "2020-11-05" -from model_helpers import set_and_check_pred_frames +from model_modules.video_prediction.models.model_helpers import set_and_check_pred_frames import tensorflow as tf from model_modules.video_prediction.layers import layer_def as ld from model_modules.video_prediction.layers.BasicConvLSTMCell import BasicConvLSTMCell diff --git a/video_prediction_tools/model_modules/video_prediction/models/vanilla_vae_model.py b/video_prediction_tools/model_modules/video_prediction/models/vanilla_vae_model.py index bc4516d9..3e74b23b 100644 --- a/video_prediction_tools/model_modules/video_prediction/models/vanilla_vae_model.py +++ b/video_prediction_tools/model_modules/video_prediction/models/vanilla_vae_model.py @@ -3,7 +3,7 @@ __email__ = "b.gong@fz-juelich.de" __author__ = "Bing Gong" __date__ = "2020-09-01" -from model_helpers import set_and_check_pred_frames +from model_modules.video_prediction.models.model_helpers import set_and_check_pred_frames import tensorflow as tf from model_modules.video_prediction.layers import layer_def as ld from tensorflow.contrib.training import HParams -- GitLab