Open wangwangwang978 opened 10 months ago
你好,默认不提供通过设置参数来保存模型。如需保存模型,建议继承类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, ...) `
你好,默认不提供通过设置参数来保存模型。如需保存模型,建议继承类flgo.experiment.logger.BasicLogger并重写log_once函数,它每隔数轮(eval_interval)会被执行一次 ,用于评估当前模型,这里头也适用于写模型参数的保存部分,例如: `class MyLogger(BasicLogger): def log_once(self):
保存模型
flgo.init(...,Logger=MyLogger, ...) `