Ross Wightman 833066b540 A few minor things in SplitBN

2020-01-05 20:07:03 -08:00

..

__init__.py

Add SplitBatchNorm. AugMix, Rand/AutoAugment, Split (Aux) BatchNorm, Jensen-Shannon Divergence, RandomErasing all working together

2020-01-05 19:58:59 -08:00

activations.py

Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo

2019-11-28 17:04:35 -08:00

adaptive_avgmax_pool.py

…

conv2d_layers.py

Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo

2019-11-28 17:04:35 -08:00

densenet.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

dla.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

dpn.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

efficientnet_blocks.py

Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo

2019-11-28 17:04:35 -08:00

efficientnet_builder.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

efficientnet.py

New PyTorch trained EfficientNet-B2 weights with my RandAugment impl

2019-12-04 11:09:47 -08:00

factory.py

Update EfficientNet comments, MobileNetV3 non-TF create fns, fix factory arg checks, bump PyTorch version req to 1.2

2019-11-28 17:43:00 -08:00

feature_hooks.py

Bring EfficientNet and MobileNetV3 up to date with my gen-efficientnet repo

2019-11-28 17:04:35 -08:00

gluon_resnet.py

ResNet updates:

2019-11-28 17:04:35 -08:00

gluon_xception.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

helpers.py

Add CondConv support for EfficientNet into WIP for GenEfficientNet Feature extraction setup

2019-11-28 17:00:52 -08:00

hrnet.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

inception_resnet_v2.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

inception_v3.py

…

inception_v4.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

median_pool.py

…

mobilenetv3.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

nasnet.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

pnasnet.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

registry.py

…

res2net.py

Fix Mobilenet V3 model name for sotabench. Minor res2net cleanup.

2019-09-05 15:47:56 -07:00

resnet.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

senet.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

split_batchnorm.py

A few minor things in SplitBN

2020-01-05 20:07:03 -08:00

test_time_pool.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00

xception.py

Finish with HRNet, weights and models updated. Improve consistency in model classifier/global pool treatment.

2019-11-29 17:56:36 -08:00