PyTorch笔记详解大全(持续更新ing...)

简介: PyTorch笔记详解大全(持续更新ing...)

1. Automatic Mixed Precision examples


2. Autograd mechanics


原文档


2.1 Excluding subgraphs from backward

Tensor属性requires_grad。

Tensor生成时requires_grad一般默认为False。

在计算图中,只有所有叶节点的requires_grad都是False时,整个树才不计算梯度。

在需要冻结部分模型或者确知某些参数不需要计算梯度时可以设置requires_grad为False,比如使用预训练模型时可以冻结其他参数、只计算仿射层参数的梯度并更新参数,输出也需要梯度。

示例代码:

model = torchvision.models.resnet18(pretrained=True)
for param in model.parameters():
    param.requires_grad = False
# Replace the last fully-connected layer
# Parameters of newly constructed modules have requires_grad=True by default
model.fc = nn.Linear(512, 100)
# Optimize only the classifier
optimizer = optim.SGD(model.fc.parameters(), lr=1e-2, momentum=0.9)


注意示例代码中注释说新构建的网络层参数会默认设置requires_grad为True,因此在真实实验中整个网络的参数往往是需要计算梯度的,所以训练时可以正常使用优化器,而在验证和测试时需要冻结参数


3. Broadcasting semantics


4. CPU threading and TorchScript inference


5. CUDA semantics


6. Distributed Data Parallel


7. Extending PyTorch


7.1 Extending torch.autograd


7.2 Extending torch.nn


7.3 Extending torch


7.4 Writing custom C++ extensions


7.5 Writing custom C extensions


8. Frequently Asked Questions


9. Features for large-scale deployments


10. Modules


11. Multiprocessing best practices


12. Reproducibility可复现性


12.1 Controlling sources of randomness

12.1.1 PyTorch random number generator

示例代码:

import torch
torch.manual_seed(0)


12.1.2 Python

示例代码:

inport random
random.seed(0)


12.1.3 Random number generators in other libraries

NumPy示例代码:

import numpy as np
np.random.seed(0)


注意,有些应用和库要用到NumPy Random Generator object而不是这个global RNG,总之情况还挺复杂。


12.1.4 CUDA convolution benchmarking

使用torch.backends.cudnn.benchmark = False来禁用这个过程,可能会使结果变差。


12.2 Avoiding nondeterministic algorithms

12.2.1 CUDA convolution determinism


12.2.2 CUDA RNN and LSTM


12.3 DataLoader

相关文章
|
1月前
|
并行计算 PyTorch TensorFlow
Ubuntu安装笔记(一):安装显卡驱动、cuda/cudnn、Anaconda、Pytorch、Tensorflow、Opencv、Visdom、FFMPEG、卸载一些不必要的预装软件
这篇文章是关于如何在Ubuntu操作系统上安装显卡驱动、CUDA、CUDNN、Anaconda、PyTorch、TensorFlow、OpenCV、FFMPEG以及卸载不必要的预装软件的详细指南。
3274 3
|
1月前
|
机器学习/深度学习 算法 PyTorch
深度学习笔记(十三):IOU、GIOU、DIOU、CIOU、EIOU、Focal EIOU、alpha IOU、SIOU、WIOU损失函数分析及Pytorch实现
这篇文章详细介绍了多种用于目标检测任务中的边界框回归损失函数,包括IOU、GIOU、DIOU、CIOU、EIOU、Focal EIOU、alpha IOU、SIOU和WIOU,并提供了它们的Pytorch实现代码。
157 1
深度学习笔记(十三):IOU、GIOU、DIOU、CIOU、EIOU、Focal EIOU、alpha IOU、SIOU、WIOU损失函数分析及Pytorch实现
|
2月前
|
机器学习/深度学习
小土堆-pytorch-神经网络-损失函数与反向传播_笔记
在使用损失函数时,关键在于匹配输入和输出形状。例如,在L1Loss中,输入形状中的N代表批量大小。以下是具体示例:对于相同形状的输入和目标张量,L1Loss默认计算差值并求平均;此外,均方误差(MSE)也是常用损失函数。实战中,损失函数用于计算模型输出与真实标签间的差距,并通过反向传播更新模型参数。
|
机器学习/深度学习 编解码 自然语言处理
【18】Vision Transformer:笔记总结与pytorch实现
【18】Vision Transformer:笔记总结与pytorch实现
853 0
【18】Vision Transformer:笔记总结与pytorch实现
|
机器学习/深度学习 数据采集 PyTorch
pytorch笔记:Dataset 和 DataLoader
pytorch笔记:Dataset 和 DataLoader
298 0
|
机器学习/深度学习 并行计算 PyTorch
PyTorch无GPU加速版本windows笔记
PyTorch无GPU加速版本windows笔记
210 0
PyTorch无GPU加速版本windows笔记
|
机器学习/深度学习 存储 算法
PyTorch函数中文文档详细解释及示例(持续更新)
PyTorch函数中文文档详细解释及示例(持续更新)
322 0
PyTorch函数中文文档详细解释及示例(持续更新)
|
机器学习/深度学习 PyTorch 算法框架/工具
【pytorch深度学习实践】笔记—03.2.随机梯度下降
【pytorch深度学习实践】笔记—03.2.随机梯度下降
136 0
【pytorch深度学习实践】笔记—03.2.随机梯度下降
|
机器学习/深度学习 PyTorch 算法框架/工具
【pytorch深度学习实践】笔记—02.线性模型
【pytorch深度学习实践】笔记—02.线性模型
200 0
【pytorch深度学习实践】笔记—02.线性模型
|
机器学习/深度学习 算法 PyTorch
【pytorch深度学习实践】笔记—03-1.梯度下降算法
【pytorch深度学习实践】笔记—03-1.梯度下降算法
215 0
【pytorch深度学习实践】笔记—03-1.梯度下降算法

热门文章

最新文章