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 9bcd65181b Add exponential moving average for model weights + few other additions and cleanup ... * ModelEma class added to track an EMA set of weights for the model being trained * EMA handling added to train, validation and clean_checkpoint scripts * Add multi checkpoint or multi-model validation support to validate.py * Add syncbn option (APEX) to train script for experimentation * Cleanup interface of CheckpointSaver while adding ema functionality		5 years ago
..
__init__.py	Uniform pretrained model handling.	6 years ago
adaptive_avgmax_pool.py	Fix pooling in mnasnet, more sensible default for AMP opt level	6 years ago
conv2d_same.py	Add support for tflite mnasnet pretrained weights and included spnasnet pretrained weights of my own.	6 years ago
densenet.py	Cleanup model_factory imports, consistent __all__ for models, fixed inception_v4 weight url	6 years ago
dpn.py	Cleanup model_factory imports, consistent __all__ for models, fixed inception_v4 weight url	6 years ago
gen_efficientnet.py	Ported Tensorflow pretrained EfficientNet weights and some model cleanup	6 years ago
gluon_resnet.py	Cleanup model_factory imports, consistent __all__ for models, fixed inception_v4 weight url	6 years ago
helpers.py	Add exponential moving average for model weights + few other additions and cleanup	5 years ago
inception_resnet_v2.py	Cleanup model_factory imports, consistent __all__ for models, fixed inception_v4 weight url	6 years ago
inception_v3.py	Cleanup model_factory imports, consistent __all__ for models, fixed inception_v4 weight url	6 years ago
inception_v4.py	Cleanup model_factory imports, consistent __all__ for models, fixed inception_v4 weight url	6 years ago
median_pool.py	Initial commit, puting some ol pieces together	6 years ago
model_factory.py	Ported Tensorflow pretrained EfficientNet weights and some model cleanup	6 years ago
pnasnet.py	Cleanup model_factory imports, consistent __all__ for models, fixed inception_v4 weight url	6 years ago
resnet.py	Replace ResNet-34 default weights with a great result from my experiments.	6 years ago
senet.py	Missed a few models in the model/_all_ list for senet	6 years ago
test_time_pool.py	Add per model crop pct, interpolation defaults, tie it all together	6 years ago
xception.py	Cleanup model_factory imports, consistent __all__ for models, fixed inception_v4 weight url	6 years ago