报错(interrupted by signal 11: SIGSEGV
(interrupted by signal 11: SIGSEGV
准备做的是数据不平衡的问题,使用的是pytorch 交叉熵函数,weight参数一维tensor。
未注意到test的交叉熵损失函数不需要.cuda()形式,所以之跑出一个train就报错了
改为: weight_three = torch.FloatTensor([0.362, 4.752, 41.809])
criterion = torch.nn.CrossEntropyLoss(weight=weight_three.cud
原创
2021-07-12 15:25:38 ·
3522 阅读 ·
0 评论