torch>=1.4.0
onnx>=1.6.0
torchvision>=0.9.0
