diff --git a/video_prediction_tools/model_modules/model_architectures.py b/video_prediction_tools/model_modules/model_architectures.py index 5836ab9fce48692252a4dbc44415b4a4f9e2c2c3..b8793ad84ee1053e2d98c43941fcb6f6a7ee8088 100644 --- a/video_prediction_tools/model_modules/model_architectures.py +++ b/video_prediction_tools/model_modules/model_architectures.py @@ -6,15 +6,10 @@ def known_models(): """ model_mappings = { 'ground_truth': 'GroundTruthVideoPredictionModel', - 'repeat': 'RepeatVideoPredictionModel', 'savp': 'SAVPVideoPredictionModel', - 'dna': 'DNAVideoPredictionModel', - 'sna': 'SNAVideoPredictionModel', - 'sv2p': 'SV2PVideoPredictionModel', 'vae': 'VanillaVAEVideoPredictionModel', 'convLSTM': 'VanillaConvLstmVideoPredictionModel', 'mcnet': 'McNetVideoPredictionModel', - 'gan': "VanillaGANVideoPredictionModel", 'convLSTM_gan': "ConvLstmGANVideoPredictionModel", 'ours_vae_l1': 'SAVPVideoPredictionModel', 'ours_gan': 'SAVPVideoPredictionModel',