qq_38504109的博客

pytorch报错：ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,256,1,1])通过查阅之后：解决方法1.在torch.utils.data.DataLoader类中或自己创建的继承于DataLoader的类中设置参数drop_last=True，把不够一个batch_size的数据丢弃。（实际操作不行）解决方法2，batch_size大于等于2分析：感

问题所在：vs code 终端安装numpy给环境安装模块应该去win+r,cmd安装打开对应的虚拟环境activate pytorch-gpu报错消除

程序运行结束后显存不足ubuntu 16.04训练网络模型时提示显存不足nvidia-smisudo kill -9 24175