解决:RuntimeError: CUDA error: device-side assert triggered

主要是网络那部分的问题,我之前是二分类,在全连接层那块

self.out = nn.Linear(10, 2)

但是还数据了,分21类,需要将全连接层输入大于分类的个数,否则报错,于是修改为

self.out = nn.Linear(64, 21)

解决

你可能感兴趣的:(python)