You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Ross Wightman
e9f1376cde
Cleanup resolve data config fns, add 'model' variant that takes model as first arg, make 'args' arg optional in original fn
|
2 years ago |
| .. |
|
data
|
Cleanup resolve data config fns, add 'model' variant that takes model as first arg, make 'args' arg optional in original fn
|
2 years ago |
|
layers
|
Update ClassifierHead module, add reset() method, update in_chs -> in_features for consistency
|
2 years ago |
|
loss
|
Remove inplace operators when calculating the loss
|
2 years ago |
|
models
|
Push all MaxxViT weights to HF hub, cleanup impl, add feature map extraction support and prompote to 'std' architecture. Fix norm head for proper embedding / feat map output. Add new in12k + ft 1k weights.
|
2 years ago |
|
optim
|
Add Adan to __init__.py
|
2 years ago |
|
scheduler
|
Scheduler update, add v2 factory method, support scheduling on updates instead of just epochs. Add LR to summary csv. Add lr_base scaling calculations to train script. Fix #1168
|
2 years ago |
|
utils
|
Update batchnorm freezing to handle NormAct variants, Add GroupNorm1Act, update BatchNormAct2d tracing change from PyTorch
|
2 years ago |
|
__init__.py
|
Major module / path restructure, timm.models.layers -> timm.layers, add _ prefix to all non model modules in timm.models
|
2 years ago |
|
version.py
|
Version 0.8.6dev0
|
2 years ago |
