问题描述

RuntimeError: Input type (torch.FloatTensor) and weight type
(torch.cuda.FloatTensor) should be the same

解决方法

方法1
device = torch.device('cuda:0')
inputs = inputs.to(device)
方法2
inputs = inputs.cuda()
Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐