From 7d0c57aed3531d2b70a06c1adbdd231378642ce2 Mon Sep 17 00:00:00 2001 From: BING GONG <b.gong@fz-juelich.de> Date: Tue, 7 Mar 2023 12:08:16 +0100 Subject: [PATCH] update tf2 --- .../model_modules/video_prediction/models/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/video_prediction_tools/model_modules/video_prediction/models/__init__.py b/video_prediction_tools/model_modules/video_prediction/models/__init__.py index 10c5c72e..e066c06a 100644 --- a/video_prediction_tools/model_modules/video_prediction/models/__init__.py +++ b/video_prediction_tools/model_modules/video_prediction/models/__init__.py @@ -7,7 +7,7 @@ from .vanilla_convLSTM_model import VanillaConvLstmVideoPredictionModel from .test_model import TestModelVideoPredictionModel from model_modules.model_architectures import known_models from .convLSTM_GAN_model import ConvLstmGANVideoPredictionModel -from .weatherBench3DCNN import WeatherBenchModel +from .weatherBench3Dcnn import WeatherBenchModel def get_model_class(model): model_mappings = known_models() -- GitLab