Ross Wightman
|
dc1bb05e8e
|
Punch cache_dir through model factory / builder / pretrain helpers. Improve some annotations in related code.
|
2024-12-06 10:25:06 -08:00 |
|
Ross Wightman
|
a7b0bfc376
|
exception for unknown arg should have been TypeError
|
2024-08-12 16:09:36 -07:00 |
|
Ross Wightman
|
bd0f79153b
|
Add weights only flag to avoid warning, try to keep bwd compat. Default to True for remote load of pretrained weights, keep False for local checkpoing load to avoid training checkpoint breaks.. fix #2249
|
2024-08-12 11:41:41 -07:00 |
|
Ross Wightman
|
0cbf4fa586
|
_orig_mod still causing issues even though I thought it was fixed in pytorch, add unwrap / clean helpers
|
2024-07-19 11:03:45 -07:00 |
mjamroz
|
7a6369156f
|
avoid getting undefined
|
2023-10-22 21:36:23 -07:00 |
|
Ross Wightman
|
041de79f9e
|
Fix numel use in helpers for checkpoint remap
|
2023-03-20 09:36:48 -07:00 |
|
Ross Wightman
|
572f05096a
|
Swin and FocalNet weights on HF hub. Add model deprecation functionality w/ some registry tweaks.
|
2023-03-18 14:55:09 -07:00 |
|
Ross Wightman
|
d0b45c9b4d
|
Make safetensor import option for now. Improve avg/clean checkpoints ext handling a bit (more consistent).
|
2023-02-18 16:06:42 -08:00 |
testbot
|
8470e29541
|
Add support to load safetensors weights
|
2023-02-16 16:20:31 -08:00 |
|
Ross Wightman
|
cda39b35bd
|
Add a deprecation phase to module re-org
|
2022-12-09 14:39:45 -08:00 |
|
Ross Wightman
|
927f031293
|
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
|
2022-12-06 15:00:06 -08:00 |