torch>=1.7.0
colorama>=0.4.4
