Adding MobileNetV3 Small pre-trained model

[datumbox]

MobileNetV3 Small

The pre-trained model was trained:

python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\
--model mobilenet_v3_small --epochs 600 --opt rmsprop --batch-size 128 --lr 0.064\ 
--wd 0.00001 --lr-step-size 2 --lr-gamma 0.973 --auto-augment imagenet --random-erase 0.2

Submitted batch job 35753749

Then we took the 3 last checkpoints (epochs 549, 528, 408) that improved the Acc@1 and averaged their parameters using the following script:

# Using the store_model_weights method from b4e56df

def avg(epochs, filename):
    paths = ["model_{}.pth".format(i) for i in epochs]
    weights = average_checkpoints(paths)
    torch.save(weights, filename.format(len(epochs)))

avg([563, 555, 554], "model_best{}avg.pth")

Validated with:

python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py\
 --model mobilenet_v3_small --test-only --pretrained

Submitted batch job 35803060

Accuracy metrics:
Acc@1 67.668 Acc@5 87.402

Speed Benchmark: 0.0165 sec per image on CPU

[fmassa]

Thanks!