Solved by eellison in post #2 
              
              
                Hi, @G.M, this is fixed in our most recent release (1.4). Could you try updating your pytorch version and giving it another go?