From 8c29a0cea28804fa1392664818fcde352eb7d797 Mon Sep 17 00:00:00 2001
From: masak1112 <gongbing1112@gmail.com>
Date: Thu, 9 Jun 2022 10:37:31 +0200
Subject: [PATCH] fix the bugs for integrating weatherbench3DCNN architecture
---
video_prediction_tools/model_modules/model_architectures.py | 2 +-
.../video_prediction/models/weatherBench3DCNN.py | 5 +++--
2 files changed, 4 insertions(+), 3 deletions(-)
diff --git a/video_prediction_tools/model_modules/model_architectures.py b/video_prediction_tools/model_modules/model_architectures.py
index b33ed8c5..79c4b5c6 100644
--- a/video_prediction_tools/model_modules/model_architectures.py
+++ b/video_prediction_tools/model_modules/model_architectures.py
@@ -13,7 +13,7 @@ def known_models():
'convLSTM_gan': "ConvLstmGANVideoPredictionModel",
'ours_vae_l1': 'SAVPVideoPredictionModel',
'ours_gan': 'SAVPVideoPredictionModel',
- "weatherBench": "WeatherBenchModel "
+ "weatherBench": "WeatherBenchModel"
}
return model_mappings
diff --git a/video_prediction_tools/model_modules/video_prediction/models/weatherBench3DCNN.py b/video_prediction_tools/model_modules/video_prediction/models/weatherBench3DCNN.py
index 725e4138..ca7b6f0b 100644
--- a/video_prediction_tools/model_modules/video_prediction/models/weatherBench3DCNN.py
+++ b/video_prediction_tools/model_modules/video_prediction/models/weatherBench3DCNN.py
@@ -13,7 +13,7 @@ from model_modules.video_prediction.losses import *
class WeatherBenchModel(object):
- def __init__(self, hparams_dict=None):
+ def __init__(self, hparams_dict=None,**kwargs):
"""
This is class for building weahterBench architecture by using updated hparameters
args:
@@ -67,7 +67,8 @@ class WeatherBenchModel(object):
loss_fun = "mse",
shuffle_on_val= True,
filter = 4,
- kernels = 4
+ kernels = 4,
+
)
return hparams
--
GitLab