Flowvision V0.1.0 Stable Release
New Features
- Support
trunc_normal_
inflowvision.layers.weight_init
#92 - Support DeiT model #115
- Support
PolyLRScheduler
andTanhLRScheduler
inflowvision.scheduler
#85 - Add
resmlp_12_224_dino
model and pretrained weight #128 - Support ConvNeXt model #93
- Add ReXNet weights #132
Bug Fixes
- Fix
F.normalize
usage in SSD #116 - Fix bug in
EfficientNet
andRes2Net
#122 - Fix error pretrained weight usage in
vit_small_patch32_384
andres2net50_48w_2s
#128
Improvements
- Refator
trunc_normal_
andlinspace
usage in Swin-T, Cross-Former, PVT and CSWin models #100 - Refator
Vision Transformer
model #115 - Refine
flowvision.models.ModelCreator
to supportModelCreator.model_list
func #123 - Refator README #124
- Refine
load_state_dict_from_url
inflowvision.models.utils
to 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 Transformer
docs #115 - Add
Getting Started
docs #124 - Add
resmlp_12_224_dino
docs #128 - Fix
VGG
docs bug #128 - Add
ConvNeXt
docs #93
Contributors
A total of 5 developers contributed to this release. Thanks @rentainhe, @simonJJJ, @kaijieshi7, @lixiang007666, @Ldpe2G