Incorrect Model runs while on GPU!

You are most likely running into a known and already fixed issue described here, so please update your PyTorch version.

1 Like