pytorch源码阅读
文章平均质量分 92
Read the fuc... source code
luckyOverflow
专业摩托车维修师傅...
展开
-
PyTorch AMP autocast源码解析
详解PyTorch Autocast核心C++源码原创 2023-02-20 18:55:20 · 1031 阅读 · 3 评论 -
PyTorch DataLoader源码分析(三)
经过前面的铺垫,DataLoader的整体架构和依赖部件都已分析完毕:PyTorch Dataloader源码分析(一)PyTorch DataLoader源码分析(二)三、DataLoader迭代器详解这一章主要介绍DataLoader的核心部分——_SingleProcessDataLoaderIter和_MultiProcessDataLoaderIter。两者的区别顾名思义,一个用于单进程,一个用于多进程。从代码实现上看,当用户选择的num_workers等于0时,DataLoader返原创 2022-03-27 21:51:56 · 2113 阅读 · 1 评论 -
PyTorch DataLoader源码分析(二)
紧接上文https://blog.csdn.net/u013608424/article/details/123172026?spm=1001.2014.3001.5502二、DataLoader主要组件上一章通过示意图简单梳理了DataLoader内部各个组件的职责和工作流程,可以看到,dataset和sampler等组件各司其职,互相独立,给用户自定义操作以及代码维护都提供了极大的便利性。为了后续分析xxxDataLoaderIter具体实现,这一章节将会一一分析各个组件的具体实现。1、Datas原创 2022-03-08 22:52:06 · 1576 阅读 · 0 评论 -
PyTorch Dataloader源码分析(一)
torch.utils.data.Dataloader是PyTorch数据加载工具的核心类。在网络脚本中使用流程一般如下:train_loader = torch.utils.data.Dataloader(...)for input, target in train_loader: # 前向计算 output = model(input) # 计算损失 loss = loss_fn(output, target) # 反向传播 optimizer.zer原创 2022-02-27 22:39:35 · 1542 阅读 · 0 评论