×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

Conv2d_first op无法在MLU上运行 已完结 明烛天南2021-08-21 22:35:03 回复 4 查看 技术答疑 使用求助 经验交流
Conv2d_first op无法在MLU上运行
分享到:

conv2d_first Op cannot run on MLU device, start running on CPU!

Traceback (most recent call last):

  File "test.py", line 67, in <module>

    net_output = net_mlu(input_mlu)

  File "/workspace/volume/private/sdk/venv/pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in __call__

    result = self.forward(*input, **kwargs)

  File "/workspace/volume/private/sdk/venv/pytorch/lib/python3.7/site-packages/torchvision/models/inception.py", line 78, in forward

    x = self.Conv2d_1a_3x3(x)

  File "/workspace/volume/private/sdk/venv/pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in __call__

    result = self.forward(*input, **kwargs)

  File "/workspace/volume/private/sdk/venv/pytorch/lib/python3.7/site-packages/torchvision/models/inception.py", line 325, in forward

    x = self.conv(x)

  File "/workspace/volume/private/sdk/venv/pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in __call__

    result = self.forward(*input, **kwargs)

  File "/workspace/volume/private/sdk/venv/pytorch/lib/python3.7/site-packages/torch_mlu/core/quantized/modules/conv.py", line 200, in forward

    self.input_std)

ValueError: To implement conv2d_first OP on cpu


使用的是inception网络模型,MLU270,

版权所有 © 2024 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭