| .. |
|
layers
|
Add 'fast' global pool option, remove redundant SEModule from tresnet, normal one is now 'fast'
|
2020-09-02 09:11:48 -07:00 |
|
pruned
|
…
|
|
|
__init__.py
|
…
|
|
|
cspnet.py
|
Add CSPResNet50 weights, 79.6 top-1 at 256x256
|
2020-08-12 11:20:11 -07:00 |
|
densenet.py
|
More uniform treatment of classifiers across all models, reduce code duplication.
|
2020-08-03 22:18:24 -07:00 |
|
dla.py
|
More uniform treatment of classifiers across all models, reduce code duplication.
|
2020-08-03 22:18:24 -07:00 |
|
dpn.py
|
More uniform treatment of classifiers across all models, reduce code duplication.
|
2020-08-03 22:18:24 -07:00 |
|
efficientnet.py
|
Add new EfficientNet-B3 and RegNetY-3.2GF weights, both just over 82 top-1
|
2020-08-11 14:18:51 -07:00 |
|
efficientnet_blocks.py
|
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
|
2020-09-01 16:05:32 -07:00 |
|
efficientnet_builder.py
|
Merge branch 'logger' into features. Change 'logger' to '_logger'.
|
2020-07-27 18:00:46 -07:00 |
|
factory.py
|
Add global_pool (--gp) arg changes to allow passing 'fast' easily for train/validate to avoid channels_last issue with AdaptiveAvgPool
|
2020-09-02 16:13:47 -07:00 |
|
features.py
|
…
|
|
|
gluon_resnet.py
|
fix test_model_default_cfgs
|
2020-08-11 23:23:57 +09:00 |
|
gluon_xception.py
|
More uniform treatment of classifiers across all models, reduce code duplication.
|
2020-08-03 22:18:24 -07:00 |
|
helpers.py
|
Cleanup Apex vs native AMP scaler state save/load. Cleanup CheckpointSaver a bit.
|
2020-09-02 15:12:59 -07:00 |
|
hrnet.py
|
hrnet features_only pretrained weight loading issue. Fix #232.
|
2020-09-01 17:37:55 -07:00 |
|
inception_resnet_v2.py
|
More uniform treatment of classifiers across all models, reduce code duplication.
|
2020-08-03 22:18:24 -07:00 |
|
inception_v3.py
|
fix test_model_default_cfgs
|
2020-08-11 23:23:57 +09:00 |
|
inception_v4.py
|
More uniform treatment of classifiers across all models, reduce code duplication.
|
2020-08-03 22:18:24 -07:00 |
|
mobilenetv3.py
|
Fix MobileNetV3 crash with global_pool='', output consistent with other models but not equivalent due to efficient head.
|
2020-08-18 14:11:30 -07:00 |
|
nasnet.py
|
More uniform treatment of classifiers across all models, reduce code duplication.
|
2020-08-03 22:18:24 -07:00 |
|
pnasnet.py
|
More uniform treatment of classifiers across all models, reduce code duplication.
|
2020-08-03 22:18:24 -07:00 |
|
registry.py
|
…
|
|
|
regnet.py
|
Add new EfficientNet-B3 and RegNetY-3.2GF weights, both just over 82 top-1
|
2020-08-11 14:18:51 -07:00 |
|
res2net.py
|
Fix some checkpoint / model str regressions
|
2020-07-29 19:43:01 -07:00 |
|
resnest.py
|
fix test_model_default_cfgs
|
2020-08-11 23:23:57 +09:00 |
|
resnet.py
|
Updated README, add wide_resnet50_2 and seresnext50_32x4d weights
|
2020-09-03 10:45:17 -07:00 |
|
rexnet.py
|
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
|
2020-09-01 16:05:32 -07:00 |
|
selecsls.py
|
fix test_model_default_cfgs
|
2020-08-11 23:23:57 +09:00 |
|
senet.py
|
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
|
2020-09-01 16:05:32 -07:00 |
|
sknet.py
|
fix test_model_default_cfgs
|
2020-08-11 23:23:57 +09:00 |
|
tresnet.py
|
Add 'fast' global pool option, remove redundant SEModule from tresnet, normal one is now 'fast'
|
2020-09-02 09:11:48 -07:00 |
|
vovnet.py
|
…
|
|
|
xception.py
|
More uniform treatment of classifiers across all models, reduce code duplication.
|
2020-08-03 22:18:24 -07:00 |
|
xception_aligned.py
|
fix test_model_default_cfgs
|
2020-08-11 23:23:57 +09:00 |
33f8a1bf36