1.检查下你的输入shape 是否满足模型输入格式要求
2.可以基于cpu trace下,然后基于CPU 运行下trace的模型,看下网络本身是否能够正常运行
3.生成离线模型代码参看:
example_mlu = torch.randn(1, 3, in_h, in_w, dtype=torch.float) net = torch.jit.trace(net.to(device), example_mlu.to(device), check_trace=False) trace_input = torch.randn(batch_size, 3, in_h, in_w, dtype=torch.float) print(trace_input.shape) ct.save_as_cambricon(mname) net(trace_input.to(device)) ct.save_as_cambricon('')
请登录后评论