I don’t think the warning should be changed in PyTorch itself, but the user should decide, if she wants to fix or ignore it. My suggestion was to raise the warning to an error in order to figure out which line of code raises it so that you could fix it in your code.