pycharm下用pytorch跑神经网络

如图，为什么会出现这个奇怪的bug
错误信息：ValueError: batch_size should be a positive integer value, but got batch_size=4