WwZzz / easyFL

An experimental platform for federated learning.
Apache License 2.0
543 stars 92 forks source link

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

Open wangwangwang978 opened 10 months ago

WwZzz commented 9 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, ...) `