PyTorch学习笔记 (一)

最新推荐文章于 2024-02-02 22:15:00 发布

Messi-Q

最新推荐文章于 2024-02-02 22:15:00 发布

阅读量262

点赞数 1

分类专栏：深度学习 python 文章标签： pytorch 深度学习

python 同时被 2 个专栏收录

18 篇文章 0 订阅

订阅专栏

深度学习

5 篇文章 0 订阅

订阅专栏

默认只有叶子节点才保持梯度，如：

1	A = Variable(torch.ones(2), requires_grad = True)

在中间节点可以设置保持梯度，如：

A = Variable(torch.ones(2), requires_grad = True)
B = A*2 
B.retain_grad()
C = B.norm()
C.backward()
print B.grad

#outputs
Variable containing:
0.7071
0.7071
[torch.FloatTensor of size 2]

也可以设置hook输出梯度：

A = Variable(torch.ones(2), requires_grad = True)
B = A*2 
def show_grad(grad):
    print(grad)
B.register_hook(show_grad)
C = B.norm()
C.backward()
#...
#grad show be auto displayed after grad of B is generated
#outputs
Variable containing:
0.7071
0.7071
[torch.FloatTensor of size 2]

每一个非叶子节点都会保存其创建，如：

1
2
3

# not B.creator now, ref: https://github.com/pytorch/tutorials/pull/91/files
print B.grad_fn 
<MulBackward0 object at 0x7f3b0536f710>

早期PyTorch的执行模型参见：function.py、engine.py
backward只能由结果是标量的operator执行，比如：nn.CrossEntropyLoss。原因暂不明。

Variable包含一些Inplcae操作（其requires_grad不能为True），均以下划线“_”结尾，如：

A = Variable(torch.ones(2), requires_grad = False)
A.fill_(3.0)

#outputs
Variable containing:
 3
 3
[torch.FloatTensor of size 2]

Storage与Tensor的概念：Storage是实际内存（1维），Tensor是对Storage的信息描述。
自动求导原理： autograd
设置CUDA同步执行，设置环境变量：CUDA_LAUNCH_BLOCKING=1，或者使用copy_强制同步
PyTorch多进程作用和使用

PyTorch的模型存取，2种方式：
1.读取参数

#save model
torch.save(the_model.state_dict(), PATH)
 
#load model
the_model = TheModelClass(*args, **kwargs)
the_model.load_state_dict(torch.load(PATH))

2.读取模型

#save model
torch.save(the_model, PATH)

#load model
the_model = torch.load(PATH)

CrossEntropyLoss与NLLLoss不同。前者是Softmax与CrossEntropy的组合，在Loss之前不要再加Softmax；后者只是CrossEntropy，需要在Loss之前加Softmax。
默认为training状态，在test时需要设置为module.eval()或者module.train(False)
- 在训练每个batch之前记得加model.train()，训练完若干个iteration之后在验证前记得加model.eval()。否则会影响dropout和BN。
- 用F.dropout()时一定要手动设参数self.training，正确用法：F.dropout(x, 0.2, self.training)。
- reference: https://www.zhihu.com/question/67209417/answer/303214329
Tensor.unsqueeze与Tensor.view作用类似，在某个地方插入一个维度（1）
Tensor.contiguous将Tensor内可能是不同的内存块整理为一块连续的内存，如果本来就是连续的则不作操作。