pytorch pre trained models with resnet 18