tmm1's picture
remove --force-reinstall from Dockerfile to ensure correct pytorch version (#492)
e356b29 unverified