torch
睡觉不准打呼噜
小白养成日记
展开
-
Pytorch踩坑集锦
1. GPU空间充足,但训练和测试同时进行时,报空间不足,即RuntimeError: CUDA out of memory.答:很多博文建议batch改小,但是可能很多人的错误在于没有固定网络,导致测试集进入网络时保存了大量参数值,因此:model.eval()with torch.no_grad(): for k, test_data in enumerate(test_loader):https://blog.csdn.net/xiaoxifei/article/d..原创 2020-10-10 19:32:16 · 235 阅读 · 1 评论 -
soft-argmax踩坑
最近在2D human pose estimation时需要用到soft-argmax,找了几个版本的函数,都有一个问题RuntimeError: "softmax_lastdim_kernel_impl" not implemented for 'Long'一、代码如下def softargmax2d(input, beta=100): *_, h, w = input.shape input = beta*input.reshape(*_, h * w) inpu原创 2020-09-19 13:52:32 · 1207 阅读 · 0 评论