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.
pytorch-image-models/timm
Ross Wightman d1e0a4607d
Switch TFDS dataset to use INTEGER_ACCURATE jpeg decode by default
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 Add crop_mode for pretraind config / image transforms. Add support for dynamo compilation to benchmark/train/validate 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