![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
Monmoka
这个作者很懒,什么都没留下…
展开
-
Tensorflow随学随记(一)
2019.9.27mocogan代码阅读的tensorflow学习记录首先 import tensorflow as tf1.tf.summary.FileWriter 指定一个文件用于储存生成的训练数据(graph)格式 tf.summary.FileWritter(path,sess.graph)常用方法 add_summary()train_summary = sess.ru...原创 2019-09-27 20:30:18 · 173 阅读 · 0 评论 -
CUDA、cuDNN、pytorch环境搭建
搭建环境:LGI1.安装python3.8+Anaconda2.安装CUDA+cuDNN显卡配置:GTX3090driver:456.71需安装CUDA版本:11.1cuDNN:8.0.5安装过程检查环境变量:测试CUDA:测试cuDNN:3.安装Pytorch首先配置清华园镜像:较全的教程1.显示已有channelconda config --show channels2.删除channelconda config --remove channels https:原创 2021-03-29 18:43:47 · 57 阅读 · 0 评论 -
transforms、ImageFolder读取图片数据
1.transforms用法from torchvision import transformstransform = transforms.Compose([transforms.Resize((224,224)), transforms.ToTensor(), ...原创 2019-10-12 21:56:52 · 1750 阅读 · 0 评论 -
Pytorch学习(一):Variable / requires_grad / .item() / .max()函数/view_as函数 /.cpu()
Variable / requires_grad / .item() / .max()函数requires_grad函数VariableVariable的功能已经改变支持Variable(Tensor)和Variable(Tensor,requires_grad)两种,返回的是一个Tensor而非Variable值.item()返回Tensor类型的数字值,此Tensor只能有一个...原创 2019-08-05 17:22:26 · 648 阅读 · 0 评论 -
视图view()函数
原创 2019-08-05 20:41:35 · 10 阅读 · 0 评论 -
Pytorch常见问题(一):MNIST集output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]
主要原因:MNIST集中的图片为灰度图片,只有一个通道,而transform = transforms.Compose([transforms.ToTensor(),transforms.Normalize(mean = (0.5,0.5,0.5),std = (0.5,0.5,0.5))])代码中mean=(0.5,0.5,0.5)则为三个通道的均值,故不匹配。【但是大多数学习P...原创 2019-09-07 12:33:40 · 8461 阅读 · 3 评论