richard:

I don’t know what version of pytorch you’re using, but I tested this on a build from source and it doesn’t error out


Thank for the reply. Did you testes by applying dataparallel on multiple GPUs?