From 5e1f193ed37f98bf6fc34af543319f28599a4625 Mon Sep 17 00:00:00 2001
From: gong1 <b.gong@fz-juelich.de>
Date: Sun, 13 Jun 2021 12:03:07 +0200
Subject: [PATCH] remove not current use models from model_architecures.py
---
video_prediction_tools/model_modules/model_architectures.py | 5 -----
1 file changed, 5 deletions(-)
diff --git a/video_prediction_tools/model_modules/model_architectures.py b/video_prediction_tools/model_modules/model_architectures.py
index 5836ab9f..b8793ad8 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',
--
GitLab