• Bar's avatar
    Add ShuffleNet v2 (#849) · 7a4845a9
    Bar authored
    * Add ShuffleNet v2
    
    Added 4 configurations: x0.5, x1, x1.5, x2
    Add 2 pretrained models: x0.5, x1
    
    * fix lint
    
    * Change globalpool to torch.mean() call
    7a4845a9
models.rst 5.12 KB