search

pytorch / vision

Datasets, Transforms and Models specific to Computer Vision
https://pytorch.org/vision
BSD 3-Clause "New" or "Revised" License
16.15k stars 6.94k forks source link

pretrained model #1588

Closed zhu2014yi closed 4 years ago

zhu2014yi commented 4 years ago

Anybody know how to train a pretrain model(etc mobile net v2 in pysot ) ?

fmassa commented 4 years ago

@zhu2014yi commands for training a mobilenet v2 in PyTorch on ImageNet can be found in https://github.com/pytorch/vision/tree/master/references/classification#mobilenetv2

This is the exact same code that we used to obtain the pre-trained weights for MobileNetV2 that are available in the modelzoo

zhu2014yi commented 4 years ago

@zhu2014yi commands for training a mobilenet v2 in PyTorch on ImageNet can be found in https://github.com/pytorch/vision/tree/master/references/classification#mobilenetv2

This is the exact same code that we used to obtain the pre-trained weights for MobileNetV2 that are available in the modelzoo

thanks