torch>=1.2.0

[extras]
PyYAML
