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
ec6921fcb0
MaxVit, ViT, ConvNeXt, and EfficientNet-v2 updates
...
* Add support for TF weights and modelling specifics to MaxVit (testing ported weights)
* More fine-tuned CLIP ViT configs
* ConvNeXt and MaxVit updated to new pretrained cfgs use
* EfficientNetV2, MaxVit and ConvNeXt high res models use squash crop/resize
2 years ago
..
data
Switch TFDS dataset to use INTEGER_ACCURATE jpeg decode by default
2 years ago
loss
Remove inplace operators when calculating the loss
2 years ago
models
MaxVit, ViT, ConvNeXt, and EfficientNet-v2 updates
2 years ago
optim
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
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
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
__init__.py
Initial multi-weight support, handled so old pretraing config handling co-exists with new tags.
2 years ago
version.py
Version 0.7.dev0 for master
2 years ago