pytorch-image-models/timm
Ross Wightman 4bc103f504 Fix CUDA crash w/ channels-last + CSP models. Remove use of chunk() 2021-02-23 13:15:52 -08:00
..
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. 2021-02-06 16:30:02 -08:00
loss asymmetric_loss 2020-10-16 17:12:28 +03:00
models Fix CUDA crash w/ channels-last + CSP models. Remove use of chunk() 2021-02-23 13:15:52 -08:00
optim Configure create_optimizer with args.opt_args 2020-12-08 00:03:09 -06:00
scheduler
utils Initial AGC impl. Still testing. 2021-02-15 23:22:44 -08:00
__init__.py
version.py Update README for AGC and bump version to 0.4.4 2021-02-16 09:13:03 -08:00