1、torch.sigmoid
2、lstm的hidden_size先改成很小的,后来又增上去;
3、最最根本的原因是:cpu和cuda之间的数据传输太多,导致很慢很慢!!!
true=label.cuda()
pred=torch.sigmoid(pred).cuda()
loss = criterion(pred.float(), true.float())
改成
1、torch.sigmoid
2、lstm的hidden_size先改成很小的,后来又增上去;
3、最最根本的原因是:cpu和cuda之间的数据传输太多,导致很慢很慢!!!
true=label.cuda()
pred=torch.sigmoid(pred).cuda()
loss = criterion(pred.float(), true.float())
改成