mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
488 Commits
138 Branches
65 Tags
47 MiB
Commit Graph

8 Commits (68fd8a267b7ee338dcf96df8d9d2c7a8b85363df)

Author SHA1 Message Date
Ross Wightman 4e61c6a12d Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript 2020-07-20 16:10:31 -07:00
Ross Wightman 9eba134d79 More models supporting feature extraction, xception, gluon_xception, inception_v3, inception_v4, pnasnet, nasnet, dla. Fix DLA unused projection params. 2020-07-19 14:02:02 -07:00
Ross Wightman eb7653614f Monster commit, activation refactor, VoVNet, norm_act improvements, more 
* refactor activations into basic PyTorch, jit scripted, and memory efficient custom auto
* implement hard-mish, better grad for hard-swish
* add initial VovNet V1/V2 impl, fix #151
* VovNet and DenseNet first models to use NormAct layers (support BatchNormAct2d, EvoNorm, InplaceIABN)
* Wrap IABN for any models that use it
* make more models torchscript compatible (DPN, PNasNet, Res2Net, SelecSLS) and add tests
 2020-06-01 17:16:52 -07:00
Ross Wightman 4d13db538f Add x48d ResNext101s to test exclude for ubuntu 2020-05-21 19:13:41 -07:00
Ross Wightman 4212cd3b9f Another attempt at getting Ubuntu test runner to work 2020-05-21 18:55:10 -07:00
Ross Wightman 20329f2630 Bring down test resolutions to see if we can at least do a fwd on the L2 models 2020-05-21 16:49:46 -07:00
Ross Wightman 3873ea710e Minor test change 2020-05-21 15:51:47 -07:00
Ross Wightman afb6bd0669 Add backward and default_cfg tests and fix a few issues found. Fix #153 2020-05-21 15:30:58 -07:00