pretrain model selection?

[math-yyj]

what kind of pretrain model was use in training? imagenet resnet pretrain or else?

[swordlidev]

Hi @mathyyj imagenet resnet pretrained model. This pretrained model can be imported by torchvision.