该repository的目的：
- 复现了相关 轻量级模型，目前已实现MobileViT模型， 以及分布式训练的范式
- 快速图像解码和高效数据预加载，加速训练速度
- 分布式训练目前实现了pytorch的分布式训练
To Do：
- 复现 MobileViT V2模型，MobileViT V3模型 以及 NextViT模型等
- 后续通过加入Nvidia的Dali框架加载图像训练数据，加速训练
- 实现基于hugging face 的 accelerate的分布式训练
-
Notifications
You must be signed in to change notification settings - Fork 0
zephyr737/classification
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published