Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
664 Commits
41 Branches
41 Tags
28 MiB
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'c16e965037'
${ noResults }
Commit Graph

13 Commits (c16e9650371d167dcb38669aa1280caba2c69dcd)

Author SHA1 Message Date
Ross Wightman 80c9d9cc72 Add 'fast' global pool option, remove redundant SEModule from tresnet, normal one is now 'fast'
4 years ago
Ross Wightman 110a7c4982 AdaptiveAvgPool2d -> mean((2,3)) for all SE/attn layers to avoid NaN with AMP + channels_last layout. See https://github.com/pytorch/pytorch/issues/43992
4 years ago
Yusuke Uchida f6b56602f9 fix test_model_default_cfgs
4 years ago
Ross Wightman 4e61c6a12d Cleanup, refactoring of Feature extraction code, add tests, fix tests, non hook feature extraction working with torchscript
4 years ago
Ross Wightman 3b9004bef9 Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces
4 years ago
Ross Wightman eb7653614f Monster commit, activation refactor, VoVNet, norm_act improvements, more
4 years ago
Ross Wightman 0ea53cecc3 Merge branch 'master' into densenet_update_and_more
4 years ago
Ross Wightman afb6bd0669 Add backward and default_cfg tests and fix a few issues found. Fix #153
4 years ago
Vyacheslav Shults a7ebe09029 Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.
4 years ago
Ross Wightman be7c784d21 Small TResNet simplification, just use SelectAdaptivePool, don't notice any perf difference
4 years ago
Ross Wightman 0004f37d25 Some TResNet cleanup.
4 years ago
talrid 8a63c1add8 finalizing
4 years ago
talrid 6209146738 TResNet models
4 years ago