You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
pytorch-image-models/timm/models
Ross Wightman 298fba09ac
Back out some activation hacks trialing upcoming pytorch changes
4 years ago
..
layers Back out some activation hacks trialing upcoming pytorch changes 4 years ago
pruned added efficientnet pruned weights 4 years ago
__init__.py Add initial impl of CrossStagePartial networks, yet to be trained, not quite the same as darknet cfgs. 4 years ago
csp.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
densenet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
dla.py Merge branch 'master' into densenet_update_and_more 4 years ago
dpn.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
efficientnet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
efficientnet_blocks.py Fix an an untested change, remove a debug print 4 years ago
efficientnet_builder.py Working on feature extraction, interfaces refined, a number of models working, some in progress. 4 years ago
factory.py Add set_layer_config contextmgr to adjust all layer configs at once, use in create_module with new args. Remove a few old warning causing constant annotations for jit. 4 years ago
feature_hooks.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
features.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
gluon_resnet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
gluon_xception.py Working on feature extraction, interfaces refined, a number of models working, some in progress. 4 years ago
helpers.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
hrnet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
inception_resnet_v2.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
inception_v3.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
inception_v4.py Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given. 4 years ago
mobilenetv3.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
nasnet.py More model feature extraction support, start to deprecate senet.py, dilations added to regnet, add proper aligned xception 4 years ago
pnasnet.py More model feature extraction support, start to deprecate senet.py, dilations added to regnet, add proper aligned xception 4 years ago
registry.py Exclude EfficientNet-L2 models from test 4 years ago
regnet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
res2net.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
resnest.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
resnet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
selecsls.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
senet.py More model feature extraction support, start to deprecate senet.py, dilations added to regnet, add proper aligned xception 4 years ago
sknet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
tresnet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
vovnet.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
xception.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago
xception_aligned.py Lots of changes to model creation helpers, close to finalizing feature extraction / interfaces 4 years ago