文章分类 -  pytorch中文官方教程

摘要:1、代码的坑 images, labels = dataiter.next() # 错误未知 是个坑 解决办法: 用 images, labels = next(dataiter)替换images, labels = dataiter.next() 成功运行! 在网上找python迭代器的写法,也没 阅读全文
posted @ 2022-11-05 23:10 JaxonYe 阅读(248) 评论(0) 推荐(0)
摘要:1、代码 import torch import torch.nn as nn import torch.nn.functional as F import numpy as np class Net(nn.Module): def __init__(self): super(Net, self). 阅读全文
posted @ 2022-11-05 18:05 JaxonYe 阅读(191) 评论(0) 推荐(0)
摘要:1、代码 """ 让我们来看一个训练步骤。 对于此示例,我们从torchvision加载了经过预训练的 resnet18 模型。 我们创建一个随机数据张量来表示具有 3 个通道的单个图像,高度&宽度为 64,其对应的label初始化为一些随机值。 """ import torch, torchvis 阅读全文
posted @ 2022-11-05 17:14 JaxonYe 阅读(580) 评论(0) 推荐(0)
摘要:https://pytorch.apachecn.org/#/ 1、代码 import torch import numpy as np data = [[1, 2], [3, 4]] print(f"data:{data}") x_data = torch.tensor(data) print(f 阅读全文
posted @ 2022-11-05 17:11 JaxonYe 阅读(96) 评论(0) 推荐(0)