[Fix] Fix torch version

[Junjun2016]

Fix #572

[codecov]

Codecov Report

Merging #670 (98b5934) into master (b9da8a1) will not change coverage.
The diff coverage is n/a.

@@           Coverage Diff           @@
##           master     #670   +/-   ##
=======================================
  Coverage   85.18%   85.18%           
=======================================
  Files         105      105           
  Lines        5670     5670           
  Branches      923      923           
=======================================
  Hits         4830     4830           
  Misses        662      662           
  Partials      178      178           

Flag	Coverage Δ
unittests	85.16% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update b9da8a1...98b5934. Read the comment docs.