Skip to content
Snippets Groups Projects
Commit 7d0c57ae authored by BING GONG's avatar BING GONG
Browse files

update tf2

parent 1b7f6bc1
Branches
Tags
No related merge requests found
Pipeline #131354 failed
...@@ -7,7 +7,7 @@ from .vanilla_convLSTM_model import VanillaConvLstmVideoPredictionModel ...@@ -7,7 +7,7 @@ from .vanilla_convLSTM_model import VanillaConvLstmVideoPredictionModel
from .test_model import TestModelVideoPredictionModel from .test_model import TestModelVideoPredictionModel
from model_modules.model_architectures import known_models from model_modules.model_architectures import known_models
from .convLSTM_GAN_model import ConvLstmGANVideoPredictionModel from .convLSTM_GAN_model import ConvLstmGANVideoPredictionModel
from .weatherBench3DCNN import WeatherBenchModel from .weatherBench3Dcnn import WeatherBenchModel
def get_model_class(model): def get_model_class(model):
model_mappings = known_models() model_mappings = known_models()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment