Ross Wightman
4efecfdc47
Add drop_connect impl to try during training, fix a few comments
2019-05-30 08:42:12 -07:00
Ross Wightman
0fc4cca2ff
Cleanup unused member with old name
2019-05-29 23:44:01 -07:00
Ross Wightman
7a6d61566e
Add EfficientNet impl, change existing depth_multipler -> channel_multiplier as definitions have been confused
2019-05-29 23:32:52 -07:00
Ross Wightman
6bff9c75dc
Cleanup model_factory imports, consistent __all__ for models, fixed inception_v4 weight url
2019-05-28 21:41:10 -07:00
Ross Wightman
a4516fe0fb
Add pretrained weights for FBNet-C
2019-05-20 11:05:25 -07:00
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
2019-05-16 22:52:17 -07:00
Ross Wightman
db8ad25a23
MobileNetV3 appears correct based on paper update, cleaned up comments and compacted last block def
2019-05-15 08:53:27 -07:00
Ross Wightman
c3fbdd4655
Fix efficient head for MobileNetV3
2019-05-11 11:23:11 -07:00
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
2019-05-11 10:23:40 -07:00
Ross Wightman
6523e4abe4
More appropriate name for se channel
2019-05-10 23:32:18 -07:00
Ross Wightman
db056d97e2
Add MobileNetV3 and associated changes hard-swish, hard-sigmoid, efficient head, etc
2019-05-10 23:28:13 -07:00
Ross Wightman
02abeb95bf
Add the url for tflite mnasnet ported weights
2019-04-28 23:21:35 -07:00
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
2019-04-28 17:45:07 -07:00
Ross Wightman
afb357ff68
Make genmobilenet weight init switchable, fix fan_out in google style linear init
2019-04-22 17:46:17 -07:00
Ross Wightman
34cd76899f
Add Single-Path NAS pixel1 model
2019-04-22 12:43:45 -07:00
Ross Wightman
419555be62
Update a few GenMobileNet comments
2019-04-22 12:30:55 -07:00
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
2019-04-21 15:54:28 -07:00
