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.

Ross Wightman 6441e9cc1b Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args.		5 years ago
..
layers	Merge branch 'master' into densenet_update_and_more	5 years ago
pruned	added efficientnet pruned weights	5 years ago
__init__.py	Add ResNeSt models	5 years ago
densenet.py	Fix memory_efficient mode for DenseNets. Add AntiAliasing (Blur) support for DenseNets and create one test model. Add lr cycle/mul params to train args.	5 years ago
dla.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
dpn.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
efficientnet.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
efficientnet_blocks.py	Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.	5 years ago
efficientnet_builder.py	Update EfficientNet feature extraction for EfficientDet. Add needed MaxPoolSame as well.	5 years ago
factory.py	Unify drop connect vs drop path under 'drop path' name, switch all EfficientNet/MobilenetV3 refs to 'drop_path'. Update factory to handle new drop args.	5 years ago
feature_hooks.py	Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo	5 years ago
gluon_resnet.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
gluon_xception.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
helpers.py	Final blurpool2d cleanup and add resnetblur50 weights, match tresnet Downsample arg order to BlurPool2d for interop	5 years ago
hrnet.py	Replace all None by nn.Identity() in HRNet modules	5 years ago
inception_resnet_v2.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
inception_v3.py	Merge branch 'master' into densenet_update_and_more	5 years ago
inception_v4.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
mobilenetv3.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
nasnet.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
pnasnet.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
registry.py	Exclude EfficientNet-L2 models from test	5 years ago
res2net.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
resnest.py	Improve dropblock impl, add fast variant, and better AMP speed, inplace, batchwise... few ResNeSt cleanups	5 years ago
resnet.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
selecsls.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
senet.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
sknet.py	Fix #139. Broken SKResNets after BlurPool addition, as a plus, SKResNets support AA now too.	5 years ago
tresnet.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago
xception.py	Replace all None by nn.Identity() in all models reset_classifier when False-values num_classes is given.	5 years ago