记录自己在torch中遇到的一些bug

向前向前向前！

已于 2022-04-12 15:12:09 修改

阅读量1.8k

点赞数

分类专栏：深度学习文章标签： opencv python 计算机视觉

于 2022-03-28 18:04:58 首次发布

本文链接：https://blog.csdn.net/weixin_44587732/article/details/123801947

版权

深度学习专栏收录该内容

17 篇文章 2 订阅

订阅专栏

记录自己在torch中遇到的一些bug

这么写会导致默认的cuda:0中一直被占用内存

FloatTensor = torch.cuda.FloatTensor if x.is_cuda else torch.FloatTensor
LongTensor  = torch.cuda.LongTensor if x.is_cuda else torch.LongTensor

anchor_w = FloatTensor(scaled_anchors_l).index_select(1, LongTensor([0]))
anchor_h = FloatTensor(scaled_anchors_l).index_select(1, LongTensor([1]))

改成：

 anchor_w = torch.from_numpy(scaled_anchors_l).type(torch.LongTensor).to(self.DEVICE).index_select(1, torch.tensor([0]).type(torch.LongTensor).to(self.DEVICE))
 anchor_h = torch.from_numpy(scaled_anchors_l).type(torch.LongTensor).to(self.DEVICE).index_select(1, torch.tensor([1]).type(torch.LongTensor).to(self.DEVICE))