• Vasilis Vryniotis's avatar
    Add MobileNetV3 architecture for Classification (#3252) · 7bf6e7b1
    Vasilis Vryniotis authored
    * Add MobileNetV3 Architecture in TorchVision (#3182)
    
    * Adding implementation of network architecture
    
    * Adding rmsprop support on the train.py
    
    * Adding auto-augment and random-erase in the training scripts.
    
    * Adding support for reduced tail on MobileNetV3.
    
    * Tagging blocks with comments.
    
    * Adding documentation, pre-trained model URL and a minor refactoring.
    
    * Handling better untrained supported models.
    7bf6e7b1
test_models.py 17.8 KB