Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

支持gan训练(多模型) #65

Open
cnstark opened this issue Aug 30, 2022 · 0 comments
Open

支持gan训练(多模型) #65

cnstark opened this issue Aug 30, 2022 · 0 comments
Labels
enhancement New feature or request

Comments

@cnstark
Copy link
Owner

cnstark commented Aug 30, 2022

整体思想:

  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 cnstark added the enhancement New feature or request label Sep 25, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

1 participant