Solved by ptrblck in post #6 
              
              
                The error message indicates that the str argument is not recognized for padding in conv2d, so you might need to update to PyTorch 1.9.0 to be able to use this argument type.