Skip to content

Commit aff25b1

Browse files
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.py
1 parent 34ee31d commit aff25b1

File tree

3 files changed

+879
-155
lines changed

3 files changed

+879
-155
lines changed

0 commit comments

Comments
 (0)