Skip to content
Snippets Groups Projects
Commit 5e1f193e authored by gong1's avatar gong1
Browse files

remove not current use models from model_architecures.py

parent 7deb6156
No related branches found
No related tags found
No related merge requests found
Pipeline #69988 passed
Stage: build
Stage: test
Stage: deploy
Show whitespace changes
Inline Side-by-side
video_prediction_tools/model_modules/model_architectures.py
+ 0
− 5
View file @ 5e1f193e
...@@ -6,15 +6,10 @@ def known_models(): ...@@ -6,15 +6,10 @@ def known_models():
""" """
model_mappings = { model_mappings = {
'ground_truth': 'GroundTruthVideoPredictionModel', 'ground_truth': 'GroundTruthVideoPredictionModel',
'repeat': 'RepeatVideoPredictionModel',
'savp': 'SAVPVideoPredictionModel', 'savp': 'SAVPVideoPredictionModel',
'dna': 'DNAVideoPredictionModel',
'sna': 'SNAVideoPredictionModel',
'sv2p': 'SV2PVideoPredictionModel',
'vae': 'VanillaVAEVideoPredictionModel', 'vae': 'VanillaVAEVideoPredictionModel',
'convLSTM': 'VanillaConvLstmVideoPredictionModel', 'convLSTM': 'VanillaConvLstmVideoPredictionModel',
'mcnet': 'McNetVideoPredictionModel', 'mcnet': 'McNetVideoPredictionModel',
'gan': "VanillaGANVideoPredictionModel",
'convLSTM_gan': "ConvLstmGANVideoPredictionModel", 'convLSTM_gan': "ConvLstmGANVideoPredictionModel",
'ours_vae_l1': 'SAVPVideoPredictionModel', 'ours_vae_l1': 'SAVPVideoPredictionModel',
'ours_gan': 'SAVPVideoPredictionModel', 'ours_gan': 'SAVPVideoPredictionModel',
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment