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 0e16d4e9fb
Add benchmark.py script, and update optimizer factory to be more friendly to use outside of argparse interface.
4 years ago
..
data Add new weights for ecaresnet26t/50t/269d models. Remove distinction between 't' and 'tn' (tiered models), tn is now t. Add test time img size spec to default cfg. 4 years ago
loss asymmetric_loss 4 years ago
models Fix CUDA crash w/ channels-last + CSP models. Remove use of chunk() 4 years ago
optim Add benchmark.py script, and update optimizer factory to be more friendly to use outside of argparse interface. 4 years ago
scheduler Merge branch 'logger' into features. Change 'logger' to '_logger'. 4 years ago
utils Initial AGC impl. Still testing. 4 years ago
__init__.py Monster commit, activation refactor, VoVNet, norm_act improvements, more 4 years ago
version.py Update README for AGC and bump version to 0.4.4 4 years ago