报错代码
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)
本文档介绍了在使用PyTorch时遇到的一个常见错误:`__init__() got an unexpected keyword argument 'data_tensor'`。问题源于新版PyTorch中对`Data.TensorDataset`构造函数的更改。正确创建TensorDataset的方式是直接传入数据和标签,而非分别指定`data_tensor`和`target_tensor`。示例代码展示了如何进行修正,帮助开发者更新他们的代码以适配最新版PyTorch。
报错代码
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)
您可能感兴趣的与本文相关的镜像
PyTorch 2.6
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
7404
3404
9639
2942

被折叠的 条评论
为什么被折叠?