from torch.autograd import Variable
自动求导
参考:
https://github.com/apachecn/pytorch-doc-zh/blob/master/docs/1.0/autograd.md
import argparse
解析参数
参考:
https://www.jianshu.com/p/aac9f8079a53
torch.manual_seed(args.seed)
#为CPU设置种子用于生成随机数,以使得结果是确定的
if args.cuda:
torch.cuda.manual_seed(args.seed)
#为当前GPU设置随机种子;
如果使用多个GPU,应该使用
torch.cuda.manual_seed_all()
为所有的GPU设置种子。
args,kwargs
https://blog.csdn.net/callinglove/article/details/45483097
原文:https://www.agiliq.com/blog/2012/06/understanding-args-and-kwargs/
model.modules()和model.children()的区别:
model.modules()和model.children()均为迭代器,model.modules()会遍历model中所有的子层,而model.children()仅会遍历当前层。
使用:
for key in model.modules():
print(key)
# model.modules()类似于 [[1, 2], 3],其遍历结果为:
[[1, 2], 3], [1, 2], 1, 2, 3
# model.children()类似于 [[1, 2], 3],其遍历结果为:
[1, 2], 3
也就是说,用model.children()进行初始化参数时,可能会漏掉部分,用model.modules()会遍历所有层
更多参考:
https://discuss.pytorch.org/t/module-children-vs-module-modules/4551
isinstance函数
https://blog.csdn.net/HS_blog/article/details/81396664
enumerate函数
https://www.runoob.com/python/python-func-enumerate.html
pytorch常用函数
https://blog.csdn.net/LemonTree_Summer/article/details/83416604
Pytorch - torch.nn.modules.upsampling 和 interpolate 函数
https://www.aiuai.cn/aifarm605.html
官网资料
https://pytorch.org/docs/master/_modules/torch/nn/modules/upsampling.html
pytorch的包
https://ptorch.com/docs/1/torchlists
.permute函数
https://blog.csdn.net/york1996/article/details/81876886
.view函数
https://blog.csdn.net/york1996/article/details/81949843
.contiguous函数
https://blog.csdn.net/appleml/article/details/80143212?utm_source=blogxgwz0
网友评论