PyTorch
文章平均质量分 79
介绍PyTorch编程知识
麻瓜1号
这个作者很懒,什么都没留下…
展开
-
PyTorch系列 | 学习率调整方法(torch.optim.lr_scheduler)
https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate原创 2021-03-25 19:05:08 · 4875 阅读 · 2 评论 -
PyTorch系列 | 参数初始化方法(权值初始化)
https://pytorch.org/docs/stable/nn.init.html原创 2021-03-25 16:44:02 · 763 阅读 · 0 评论 -
PyTorch系列 | 随机种子与可复现性
torch.cuda.manual_seed_all原创 2021-03-24 22:33:13 · 1797 阅读 · 2 评论 -
PyTorch系列 | 张量类型及tensor与ndarray的转换
from_numpy()tensor()原创 2021-03-24 21:08:01 · 1461 阅读 · 0 评论 -
PyTorch系列 | 自定义dataset(torch.utils.data.Dataset与torchvision.datasets.ImageFolder)
是原创 2021-03-05 01:08:09 · 3034 阅读 · 3 评论 -
PyTorch系列 | _, predicted = torch.max(outputs.data, 1)的理解
使用pytorch的小伙伴们,一定看过下面这段代码_, predicted = torch.max(outputs.data, 1)那么,这里的 下划线_ 表示什么意思?首先,torch.max()这个函数返回的是两个值,第一个值是具体的value(我们用下划线_表示),第二个值是value所在的index(也就是predicted)。那么,这个 下划线_ 表示的就是具体的value,也就是输出的最大值。那么为什么用 下划线_,可不可以用其他的变量名称来代替,比如x?答案自然是可以的。那么为什么这原创 2020-12-18 19:07:23 · 30942 阅读 · 17 评论 -
PyTorch系列 | correct += (predicted == labels).sum().item()的理解
各位小伙伴肯定看到过下面这段代码:correct += (predicted == labels).sum().item()这里面(predicted == labels)是布尔型,为什么可以接sum()呢?我做了个测试,如果这里的predicted和labels是列表形式就会报错,如果是numpy的数组格式,会返回一个值,如果是tensor形式,就会返回一个张量。举个例子:import torcha = torch.tensor([1,2,3])b = torch.tensor([1,3,原创 2020-12-18 20:33:08 · 13356 阅读 · 7 评论