@agolynski could you suggest how/where I should add logging?
Yes, I’m on a 2 GPU machine from LambdaLabs 1 if you want to try and replicate. (I upgrade torch to the latest release when I create the instance.)
Do I need to add anything extra to use the GPUs?