runtimeerror: expected a 'cuda' device type for generator but found 'cpu'
你数据挂载的时候出错了，cuda是gpu，可能你的电脑只有cpu