小编dag*_*sky的帖子

RuntimeError:CUDA 错误:丢失函数触发设备端断言

/pytorch/aten/src/ATen/native/cuda/Loss.cu:102:operator():块:[18,0,0],线程:[54,0,0] 断言input_val >= zero && input_val <= one失败。

/pytorch/aten/src/ATen/native/cuda/Loss.cu:102:operator():块:[18,0,0],线程:[55,0,0] 断言input_val >= zero && input_val <= one失败。

/pytorch/aten/src/ATen/native/cuda/Loss.cu:102:operator():块:[18,0,0],线程:[56,0,0] 断言input_val >= zero && input_val <= one失败。

/pytorch/aten/src/ATen/native/cuda/Loss.cu:102:operator():块:[18,0,0],线程:[57,0,0] 断言input_val >= zero && input_val <= one失败。

/pytorch/aten/src/ATen/native/cuda/Loss.cu:102:operator():块:[18,0,0],线程:[58,0,0] 断言input_val >= zero && input_val <= one失败。

/pytorch/aten/src/ATen/native/cuda/Loss.cu:102:operator():块:[18,0,0],线程:[59,0,0] 断言input_val >= zero && input_val <= one失败。

回溯(最近一次调用最后一次):
文件“run_toys.py”,第 215 行,
损失 = criteria(torch.reshape(out, [-1, dataset.out_dim]), torch.reshape(target, [-1, dataset .out_dim]))
文件“/usr/local/python3/lib/python3.6/site-packages/torch/nn/modules/module.py”,第 727 行,在 _call_impl
结果 = self.forward(*input, …

python-3.x deep-learning pytorch

4
推荐指数
1
解决办法
8147
查看次数

标签 统计

deep-learning ×1

python-3.x ×1

pytorch ×1