easytorch
easytorch copied to clipboard
支持gan训练(多模型)
整体思想:
- 多model 的 runner作为基础runner,原始的单model runner作为基础runner的子类,兼容旧代码
- config里面多个optim分别指定model,多个lrscheduler分别指定optim
- Param参数,将cfg中的变量自动添加为runner的属性。这里是为了支持多个model、optim、lrscheduler的访问。 改动:
- build model时支持build多model,通过config配置
- build optim和build lr scheduler作为成员函数,支持build多个optim和lr scheduler,通过config配置
- backward添加一个可选参数,指定optim
- 保存和加载checkpoint时,支持多model