easytorch icon indicating copy to clipboard operation
easytorch copied to clipboard

支持gan训练(多模型)

Open cnstark opened this issue 2 years ago • 0 comments

整体思想:

  1. 多model 的 runner作为基础runner,原始的单model runner作为基础runner的子类,兼容旧代码
  2. config里面多个optim分别指定model,多个lrscheduler分别指定optim
  3. Param参数,将cfg中的变量自动添加为runner的属性。这里是为了支持多个model、optim、lrscheduler的访问。 改动:
  4. build model时支持build多model,通过config配置
  5. build optim和build lr scheduler作为成员函数,支持build多个optim和lr scheduler,通过config配置
  6. backward添加一个可选参数,指定optim
  7. 保存和加载checkpoint时,支持多model

cnstark avatar Aug 30 '22 10:08 cnstark