fix bugs to save statsitics information to output directory during training process

video_prediction_tools/model_modules/video_prediction/datasets/era5_dataset.py

@@ -142,16 +142,17 @@ class ERA5Dataset(BaseDataset):
     def save_stats(self, variables: list = None, output_dir: str = None, **kwargs):
 
         output_file = os.path.join(output_dir, "statistics.json")
-        stats = {}
         stats_keys = list(kwargs.keys())
         stats_values = list(kwargs.values())
-        stats_var = {} #for each variable
+        stats = {}
         print("stats_values",stats_values)
         for i, var in enumerate(variables):
+            stats_var = {} #for each variable
             for j, key in enumerate(stats_keys):
                 print("var: {}, key:{}, value: {}".format(var, key,stats_values[j][i]))
-                stats_var.update({var:{key: float(stats_values[j][i])}})
-            stats.update(stats_var)
+                stats_var.update({key: float(stats_values[j][i])})
+                print("stats_var",stats_var)
+            stats.update({var:stats_var})
         print("stats",stats)
         #save to output directory
         json.dumps(stats)