Flowvision V0.1.0 Stable Release
New Features
- Support
trunc_normal_inflowvision.layers.weight_init#92 - Support DeiT model #115
- Support
PolyLRSchedulerandTanhLRSchedulerinflowvision.scheduler#85 - Add
resmlp_12_224_dinomodel and pretrained weight #128 - Support ConvNeXt model #93
- Add ReXNet weights #132
Bug Fixes
- Fix
F.normalizeusage in SSD #116 - Fix bug in
EfficientNetandRes2Net#122 - Fix error pretrained weight usage in
vit_small_patch32_384andres2net50_48w_2s#128
Improvements
- Refator
trunc_normal_andlinspaceusage in Swin-T, Cross-Former, PVT and CSWin models #100 - Refator
Vision Transformermodel #115 - Refine
flowvision.models.ModelCreatorto supportModelCreator.model_listfunc #123 - Refator README #124
- Refine
load_state_dict_from_urlinflowvision.models.utilsto support downloading pretrained weights to cache dir~/.oneflow/flowvision_cache#127 - Rebuild a cleaner model zoo and test all the model with pretrained weights released in flowvision #128
Docs Update
- Update
Vision Transformerdocs #115 - Add
Getting Starteddocs #124 - Add
resmlp_12_224_dinodocs #128 - Fix
VGGdocs bug #128 - Add
ConvNeXtdocs #93
Contributors
A total of 5 developers contributed to this release. Thanks @rentainhe, @simonJJJ, @kaijieshi7, @lixiang007666, @Ldpe2G