torch==1.8.1 torchvision==0.9.1 coremltools==5.0b1