weixin_43711726的博客

今天在用pytorch训练模型遇到一个错误：RuntimeError: ‘lengths’ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor解决：参考https://github.com/pytorch/pytorch/releasesinp = torch.rand(10, 2, 3, device=“cuda”)lengths = torch.tensor([10, 7], device=“cu

首先找到对应的错误代码：C:\Users\admin\PycharmProjects\untitled3\mnist_generateds.py:56: string_input_producer (from tensorflow.python.training.input) is deprecated and will be removed in a future version.Instr...