Python save_json示例

示例#1

文件： callbacks_rag.py 项目： DanielaBWeiss/transformers

 def on_test_end(self, trainer: pl.Trainer, pl_module: pl.LightningModule):
     save_json(pl_module.metrics, pl_module.metrics_save_path)
     return self._write_logs(trainer, pl_module, "test")

示例#2

文件： callbacks_rag.py 项目： DanielaBWeiss/transformers

 def on_validation_end(self, trainer: pl.Trainer, pl_module):
     save_json(pl_module.metrics, pl_module.metrics_save_path)

示例#3

文件： finetune_rag.py 项目： MaveriQ/transformers

 def save_metrics(self, latest_metrics, type_path) -> None:
     self.metrics[type_path].append(latest_metrics)
     save_json(self.metrics, self.metrics_save_path)