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
bed350f5e5
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 |
| .. |
|
data
|
Remove print from auto_augment
|
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 |
