2021年10月9日
(y_hat.argmax(dim=1) ==lable).sum().cpu().item()
摘要： print(y_hat.argmax(dim=1)) print(y_hat.argmax(dim=1) ==lable) print((y_hat.argmax(dim=1) ==lable).sum()) print((y_hat.argmax(dim=1) ==lable).sum().cpu 阅读全文
posted @ 2021-10-09 17:29 祥瑞哈哈哈 阅读(540) 评论(0) 推荐(0)
"host_softmax" not implemented for 'Long'
摘要： l=loss(y_hat,lable) 损失函数中输出和标签的位置是否写反。 阅读全文
posted @ 2021-10-09 16:53 祥瑞哈哈哈 阅读(1929) 评论(0) 推荐(0)