Impala Pytorch Baseline Bug(s)

[jyotish]

Hello @bob_wei

Can you try using pytorch 1.3.x version?

Pytorch version

Hello all, Just thought to mention that torch==1.5.0a0+8f84ded, which is standard in the Docker image nvcr.io/nvidia/pytorch:20.03-py3, doesn’t work with the starter kit. Uninstall torch and install torch==1.3.1. Hope this helps. Best, Ahmed