ICode9

精准搜索请尝试: 精确搜索
  • libtorch1.5加载lstmFC.pt模型到cuda:1,报错2022-06-15 20:31:52

    1、情景   pytorch的模型,torch.jit.trace转换成pt文件   然后通过C++加载调用模型; 2、报错内容: terminate called after throwing an instance of 'std::runtime_error' what(): Input and hidden tensors are not at the same device, found input tensor at cuda:1 and

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有