随笔分类 -  pytorch

摘要:tensor hook register_hook module hook register_forward_hook 前向传播计算 register_backward_hook 反向传播计算 Ref: http://www.tensorinfinity.com/paper_198.html 阅读全文
posted @ 2019-12-24 21:01 youqia 阅读(147) 评论(0) 推荐(0)
摘要:目的: 固定住训练的顺序等变量,使实验可复现 Ref: http://www.ishenping.com/ArtInfo/3265474.html https://www.lizenghai.com/archives/30179.html 阅读全文
posted @ 2019-12-11 14:39 youqia 阅读(754) 评论(0) 推荐(0)
摘要:https://discuss.pytorch.org/t/guidelines-for-assigning-num-workers-to-dataloader/813/5 num_workers 影响机器性能 阅读全文
posted @ 2019-03-16 19:39 youqia 阅读(1798) 评论(0) 推荐(0)
摘要:import torch import torch.utils.data as Data import torchvision import torchvision.transforms as transforms import matplotlib.pyplot as plt import numpy as np #torch.manual_seed(1) # reproducibl... 阅读全文
posted @ 2019-01-11 19:42 youqia 阅读(1154) 评论(0) 推荐(0)
摘要:1.创建一个未初始化矩阵 2.均匀分布 3.创建一个零矩阵 4.自定义初始化矩阵,覆盖并改变dtype 5.产生单位矩阵 6.将已有矩阵的元素都变成1 7.加 8.打印第一列 9.view 10.item (单元素矩阵转数字) 11.tensor转numpy 12.numpy转tensor 13.n 阅读全文
posted @ 2018-11-19 20:37 youqia 阅读(822) 评论(0) 推荐(0)