Ross Wightman
4efecfdc47
Add drop_connect impl to try during training, fix a few comments
6 years ago
Ross Wightman
0fc4cca2ff
Cleanup unused member with old name
6 years ago
Ross Wightman
7a6d61566e
Add EfficientNet impl, change existing depth_multipler -> channel_multiplier as definitions have been confused
6 years ago
Ross Wightman
6bff9c75dc
Cleanup model_factory imports, consistent __all__ for models, fixed inception_v4 weight url
6 years ago
Ross Wightman
a4516fe0fb
Add pretrained weights for FBNet-C
6 years ago
Ross Wightman
76539d905e
Some transform/data/loader refactoring, hopefully didn't break things
...
* factor out data related constants to own file
* move data related config helpers to own file
* add a variant of RandomResizeCrop that randomizes interpolation method
* remove old Numpy version of RandomErasing
* cleanup torch version of RandomErasing and use it in either GPU loader batch mode or single image cpu Transform
6 years ago
Ross Wightman
db8ad25a23
MobileNetV3 appears correct based on paper update, cleaned up comments and compacted last block def
6 years ago
Ross Wightman
c3fbdd4655
Fix efficient head for MobileNetV3
6 years ago
Ross Wightman
17da1adaca
A few MobileNetV3 tweaks
...
* fix expansion ratio on early block
* change comment re stride mistake in paper
* fix rounding not being called properly for all multipliers != 1.0
6 years ago
Ross Wightman
6523e4abe4
More appropriate name for se channel
6 years ago
Ross Wightman
db056d97e2
Add MobileNetV3 and associated changes hard-swish, hard-sigmoid, efficient head, etc
6 years ago
Ross Wightman
02abeb95bf
Add the url for tflite mnasnet ported weights
6 years ago
Ross Wightman
4663fc2132
Add support for tflite mnasnet pretrained weights and included spnasnet pretrained weights of my own.
...
* tensorflow 'SAME' padding support added to GenMobileNet models for tflite pretrained weights
* folded batch norm support (made batch norm optional and enable conv bias) for tflite pretrained weights
* add url for spnasnet1_00 weights that I recently trained
* fix SE reduction size for semnasnet models
6 years ago
Ross Wightman
afb357ff68
Make genmobilenet weight init switchable, fix fan_out in google style linear init
6 years ago
Ross Wightman
34cd76899f
Add Single-Path NAS pixel1 model
6 years ago
Ross Wightman
419555be62
Update a few GenMobileNet comments
6 years ago
Ross Wightman
bc264269c9
Morph mnasnet impl into a generic mobilenet that covers Mnasnet, MobileNetV1/V2, ChamNet, FBNet, and related
...
* add an alternate RMSprop opt that applies eps like TF
* add bn params for passing through alternates and changing defaults to TF style
6 years ago