pytorch笔记
文章平均质量分 50
记录纯小白在学习pytorch过程中遇到的问题等
ahaha_zzzz
这个作者很懒,什么都没留下…
展开
-
pytorch入门学习笔记①
pytorch入门学习不同版本的模块导入语句1. summary writerfrom tensorboardX import SummaryWriterfrom torch.utils.tensorboard import SummaryWriter2. 日志读写功能目前碰到过的使用日志的有:(1)使用python第三方库logging(2) 使用tensorflow中的方法第一种情况,参考写法:import loggingloss_log = log_creater(file原创 2021-03-20 15:29:40 · 231 阅读 · 0 评论 -
pytorch入门学习笔记②
Numpy与Torch中的size、shape、view等等那些事儿最近脑子不好使,numpy和torch中的一些关于维度的函数及其功能总是搞混。为了记忆方便以及后续复习,暂且整理了些内容。直接上图相对应的例子如下:以 [[1, 2, 3], [4, 5, 6]]为例。关于array和tensor的维度变化上的函数不止这些,遇到又犯浑的时候再补充吧~...原创 2021-03-16 18:38:12 · 282 阅读 · 0 评论 -
Windows10中安装GPU版本的Pytorch
Windows10下的Pytorch安装教程1. CUDA和cuDNN的下载和安装查看电脑CUDA版本查看Pytorch中可支持的CUDA版本1.1 下载1.2 安装1.2.1 安装CUDA1.2.2 安装cuDNN2. Pytorch安装2.1 安装包的下载2.2 Pytorch的安装首先需要说明的是,这个教程是在已经安装好Anacoda和PyCharm的前提下进行的,并且需要一个已经配置了基础库的虚拟环境。我是在cmd中通过conda的命令创建了名为pytorch_gpu01的虚拟环境,其中的Pyth原创 2020-10-04 22:04:56 · 1758 阅读 · 1 评论 -
model.eval()与with torch.no_grad()
requires_grad、volatile与no_gradrequires_gradvolatileno_gradmodel.eval()与with torch.no_grad()总结:requires_grad=True 要求计算梯度requires_grad=False 不要求计算梯度with torch.no_grad()或者@torch.no_grad()中的数据不需要计算梯度,也不会进行反向传播。(torch.no_grad()是新版本pytorch中volatile的替代)requi原创 2020-09-01 22:11:43 · 6521 阅读 · 1 评论