Commit aff25b1
authored
Update timm universal (support transformer-style model) (#1004)
* Update timm_universal.py
* Fix ruff style and typing
* Update encoders_timm.rst
* Fix typo error
* Fix typo error & Update doc
* Fix typo error
* Support channel-last format
* Update encoders_timm.rst
* Update timm_universal.py
* Fix ruff style
* Update timm_universal.py
1. rename temporary model
2. create temporary model on meta device to speed up
* Add tests/test_models & fix type
* Update test_models.py
* Update test_models.py1 parent 34ee31d commit aff25b1
File tree
3 files changed
+879
-155
lines changed- docs
- segmentation_models_pytorch/encoders
- tests
3 files changed
+879
-155
lines changed
0 commit comments