最新 最热

pytorch的python API略读--tensor(六)

torch.permute:返回输入tensor的一个维度层次的置换,不知道置换的概念,可以搜索下群论中的置换群的定义及置换的记号。用法如下:

2022-07-04
3

pytorch的python API略读--tensor(三)

上面torch.where函数返回tensor的某个元素的值遵循这样的选择:如果cvtutorials中的某个元素大于0,那么保留,否则设置为0,用数学公式表达如下:

2022-07-04
2

03-2 轻松学 PyTorch 手写字体识别 MNIST (实战—上)

本期视频内容:手写字体识别 MNIST (实战 - 上)视频地址:http://mpvideo.qpic.cn/0bc32aabyaaavealzndykvrfbugddtiaahaa.f10002

2022-06-15
2

pytorch学习笔记(三):反向传播

细节提要: Tensor张量可以视作数据结构:数据data + 梯度grad(grad也是张量) 核心代码:l.backward()反向传播,程序自动求出所有需要的梯度 w.grad.data.zero_()每步做完清零很重要,否则影响下一次求的梯度...

2022-06-14
3

英伟达TensorRT加速AI推理 Hackathon 2022初赛复盘

阿里云天池联合英伟达在4月份启动了“英伟达 TensorRT 加速 AI 推理 Hackathon2022——Transformer 模型优化赛”,助力开发者在编程实践中寻找更高效的解决路径,同时也让更多工程师通过大赛实践更好的掌握 TensorRT 这...

2022-05-27
2

深度学习Int8的部署推理原理和经验验证

论文出处:《Integer Quantization for Deep Learning Inference Principles and Empirical Evaluation》时间:2020.April单位:NVIDIA

2022-05-27
1

[源码解析] 模型并行分布式训练 Megatron (4) --- 如何设置各种并行

NVIDIA Megatron 是一个基于 PyTorch 的分布式训练框架,用来训练超大Transformer语言模型,其通过综合应用了数据并行,Tensor并行和Pipeline并行来复现 GPT3,值得我们深入分析其背后机理。...

2022-05-09
3

PyTorch 1.0 中文文档:torchvision.utils

译者:BXuan694torchvision.utils.make_grid(tensor, nrow=8, padding=2, normalize=False, range=None, scale_each=False, pad_value=0)把图片排列成网格形状。参数:tensor(Tensor 或 l...

2022-05-07
1

PyTorch 1.0 中文文档:torchvision 参考

torchvision 包收录了若干重要的公开数据集、网络模型和计算机视觉中的常用图像变换

2022-05-07
4

PyTorch 1.0 中文文档:torch.utils.dlpack

译者:kunwuztorch.utils.dlpack.from_dlpack(dlpack) → Tensor将DLPack解码成Tensor张量。参数:dlpack – 一个有着dltensor张量的PyCapsule对象这个张量会与dlpack对象共享存储空间。注意每个dlpack对象只能使用一次...

2022-05-07
1