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

Add docs for vision dataset, optim, trainer and vit models #148

Merged
merged 23 commits into from
Mar 2, 2022

Conversation

rentainhe
Copy link
Contributor

@rentainhe rentainhe commented Mar 1, 2022

TODO

这个PR主要完成的事情, 添加昨天开会分配到的相关文档, 因为只修改了docstring, 所以就放在一个PR下统一提交:

  • 视觉dataset相关的docs: mnist, cifar10, cifar100, imagenet
  • 模型相关的docstring: ViT
  • 优化器相关的docstring: optim
  • Trainer相关的文档导出
  • data部分的文档分模块导出
  • Trainer相关文档的Refine
  • 简单添加GraphTrainer文档

文档可视化(保证是最新的,修改后会把最新的图放上来)

  • Datasets相关
    image

  • ViT文档
    image

  • optim文档
    image

@rentainhe rentainhe mentioned this pull request Mar 1, 2022
3 tasks
libai/trainer/__init__.py Outdated Show resolved Hide resolved
@rentainhe rentainhe changed the title Add Docs Add docs for vision dataset, optim, trainer and vit models Mar 1, 2022
@rentainhe rentainhe requested review from oneflow-ci-bot and removed request for oneflow-ci-bot March 2, 2022 02:56
@rentainhe rentainhe removed the request for review from oneflow-ci-bot March 2, 2022 02:58
@rentainhe rentainhe requested review from oneflow-ci-bot and removed request for oneflow-ci-bot March 2, 2022 03:10
@rentainhe rentainhe merged commit 5eab022 into main Mar 2, 2022
@rentainhe rentainhe deleted the dev_rth_add_docs branch March 2, 2022 03:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants