随笔分类 -  pytorch

pytorch数据集中类型不匹配
摘要：pytorch中，出现的错误： return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing) expected 阅读全文
posted @ 2023-01-07 21:23 __sunshine 阅读(323) 评论(0) 推荐(0) 编辑
torch的数据集
摘要：torch.utils.data.TensorDataset 这个类可以初始化数据集 例子： import torch from torch.utils import data # torch.utils.data.dataset 类的使用 x = torch.arange(12, dtype=to 阅读全文
posted @ 2023-01-06 10:26 __sunshine 阅读(59) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示