pytorch
wavehaha
这个作者很懒,什么都没留下…
展开
-
段错误(核心已转储)错误及解决过程
在Linux服务器上训练模型时,每当训练数据时程序就会自动停掉,报段错误(核心已转储),经过搜索有人说是程序越界了,stack太小了,于是我用ulimit -a查看了Linux中stack的大小,然后又使用ulimit -c 10240修改stack的大小,还是不行。然后我就想,既然他说我的数组过大,那我就设置小一点,结果也还是不行。既然如此至少得找到是在哪里出了问题,然后我利用print语句,发现了出错的位置:但我还是没想通为啥这里会出错,直到我打印了state,next_state以后我发现,原创 2021-04-22 20:10:56 · 10831 阅读 · 10 评论 -
RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 ‘mat2‘ in
RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 ‘mat2’ in call to _th_mm最近在利用Linux服务器训练强化学习模型,由于是现在CPU上跑的,后来放到了GPU跑,结果出现了上述问题,大致意思是object 的 device 类型期望得到的是 cuda 类型,但是实际上的类型确实 cpu 类型经检查发现虽然device = torch.device(原创 2021-04-22 20:08:25 · 5641 阅读 · 0 评论 -
RuntimeError: Could not infer dtype of numpy.int64
RuntimeError: Could not infer dtype of numpy.int64问题:再写强化学习代码时,遇到该错误,代码如下,对于传的参数max_action=2.0,使用self.action_scale = torch.tensor(max_action)会报如上错误,主要是max_action变为了numpy.int64,所以不能用torch.tensor()class Actor(nn.Module): # def __init__(self, action_di原创 2022-05-11 15:43:27 · 4590 阅读 · 0 评论