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 |