2020 年 4月 25 日随笔档案 - 龙雪

摘要： pytorch 计算 CrossEntropyLoss 不需要经 softmax 层激活！用 pytorch 实现自己的网络时，如果使用CrossEntropyLoss 我总是将网路输出经 softmax激活层后... 阅读全文

posted @ 2020-04-25 00:39 龙雪阅读(676) 评论(4) 推荐(0) 编辑

摘要：问题：UserWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=... 阅读全文

posted @ 2020-04-25 00:32 龙雪阅读(868) 评论(0) 推荐(0) 编辑

摘要：在pytorch结合cuda下，直接使用output.data.numpy()会报标题错误：TypeError: can't convert CUDA tensor to numpy. Use Tensor.cp... 阅读全文

posted @ 2020-04-25 00:27 龙雪阅读(7252) 评论(0) 推荐(0) 编辑

龙雪的博客