From 9e172bea2038d1965e8cd97aab632facb3d1caca Mon Sep 17 00:00:00 2001
From: gong1 <b.gong@fz-juelich.de>
Date: Tue, 17 Jan 2023 17:56:30 +0100
Subject: [PATCH] remove the useless model architecture
---
video_prediction_tools/model_modules/model_architectures.py | 3 ---
1 file changed, 3 deletions(-)
diff --git a/video_prediction_tools/model_modules/model_architectures.py b/video_prediction_tools/model_modules/model_architectures.py
index 85dee99b..6c699ffb 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'
--
GitLab