Weights properly stored on GPU