diff --git a/video_prediction_tools/model_modules/model_architectures.py b/video_prediction_tools/model_modules/model_architectures.py index 85dee99bc89f896793f12e37976869a8cc24afbb..6c699ffb63b9dc9e1de9c3362359a053d5cc3844 100644 --- a/video_prediction_tools/model_modules/model_architectures.py +++ b/video_prediction_tools/model_modules/model_architectures.py @@ -1,12 +1,9 @@ def known_models(): """ An auxilary function - ours_vae_l1 and ours_gan are from savp papers :return: dictionary of known model architectures """ model_mappings = { - 'ground_truth': 'GroundTruthVideoPredictionModel', - 'savp': 'SAVPVideoPredictionModel', 'convLSTM': 'VanillaConvLstmVideoPredictionModel', 'convLSTM_gan': "ConvLstmGANVideoPredictionModel", 'weatherBench': 'WeatherBenchModel'