如何设置参数使得保存模型最优权重的问题？

[wangwangwang978]

[WwZzz]

你好，默认不提供通过设置参数来保存模型。如需保存模型，建议继承类flgo.experiment.logger.BasicLogger并重写log_once函数，它每隔数轮（eval_interval）会被执行一次 ，用于评估当前模型，这里头也适用于写模型参数的保存部分，例如： `class MyLogger(BasicLogger): def log_once(self): # 保存模型 ... torch.save(self.server.model, save_path) ...

flgo.init(...,Logger=MyLogger, ...) `