pytorch-image-models/timm
Ross Wightman 7ef7788ee9 Fix CUDA crash w/ channels-last + CSP models. Remove use of chunk() 2021-03-06 16:56:34 -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
models Fix CUDA crash w/ channels-last + CSP models. Remove use of chunk() 2021-03-06 16:56:34 -08:00
optim
scheduler
utils Initial AGC impl. Still testing. 2021-02-15 23:22:44 -08:00
__init__.py
version.py Tweak NFNet, ResNetV2, RexNet feature extraction, use pre-act features for simplicty. Update tests for pytorch 1.8. 0.4.5 version bump. 2021-03-06 12:48:06 -08:00