mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity
413 Commits 141 Branches 65 Tags
Commit Graph

5 Commits

Author SHA1 Message Date
Vyacheslav Shults
a7ebe09029 Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given. 
Make small code refactoring
 2020-05-06 09:54:03 +03:00
Ross Wightman
be7c784d21 Small TResNet simplification, just use SelectAdaptivePool, don't notice any perf difference 2020-04-27 17:50:19 -07:00
Ross Wightman
0004f37d25 Some TResNet cleanup. 
* allow use of global pool arg, test-time-pooling
* clean checkpoints to just contain state dict, add 448 res checkpoints
* support DataParallel via lazy filter creation for JIT Downsample
* some minor formatting (mostly alignment) preferences
 2020-04-27 13:21:05 -07:00
talrid
8a63c1add8 finalizing 2020-04-12 19:08:35 +03:00
talrid
6209146738 TResNet models 2020-04-12 18:44:12 +03:00