报错代码

torch_dataset = Data.TensorDataset(data_tensor=data, target_tensor=label)

报错

TypeError: __init__() got an unexpected keyword argument 'data_tensor'

原因:时代变了,新版把输入变成了可变参数,直接输入即可

正确写法

torch_dataset = Data.TensorDataset(data, label)

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐