obj=obj.cuda()
其中obj是一个字典,但是字典类型不支持直接cuda()操作。可以用下面的方法将obj的每个value都变成cuda()类型。
obj={key:obj[key].cuda() for key in obj}