随笔分类 -  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 阅读(396) 评论(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 阅读(67) 评论(0) 推荐(0)