
Pytorch
Monmoka
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
Tensorflow随学随记(一)
2019.9.27 mocogan代码阅读的 tensorflow学习记录 首先 import tensorflow as tf 1.tf.summary.FileWriter 指定一个文件用于储存生成的训练数据(graph) 格式 tf.summary.FileWritter(path,sess.graph) 常用方法 add_summary() train_summary = sess.ru...原创 2019-09-27 20:30:18 · 203 阅读 · 0 评论 -
CUDA、cuDNN、pytorch环境搭建
搭建环境:LGI 1.安装python3.8+Anaconda 2.安装CUDA+cuDNN 显卡配置:GTX3090 driver:456.71 需安装CUDA版本:11.1 cuDNN:8.0.5 安装过程 检查环境变量: 测试CUDA: 测试cuDNN: 3.安装Pytorch 首先配置清华园镜像: 较全的教程 1.显示已有channel conda config --show channels 2.删除channel conda config --remove channels https:原创 2021-03-29 18:43:47 · 264 阅读 · 0 评论 -
transforms、ImageFolder读取图片数据
1.transforms用法 from torchvision import transforms transform = transforms.Compose([transforms.Resize((224,224)), transforms.ToTensor(), ...原创 2019-10-12 21:56:52 · 1826 阅读 · 0 评论 -
Pytorch学习(一):Variable / requires_grad / .item() / .max()函数/view_as函数 /.cpu()
Variable / requires_grad / .item() / .max()函数 requires_grad函数 Variable Variable的功能已经改变 支持Variable(Tensor)和Variable(Tensor,requires_grad)两种,返回的是一个Tensor而非Variable值 .item() 返回Tensor类型的数字值,此Tensor只能有一个...原创 2019-08-05 17:22:26 · 720 阅读 · 0 评论 -
视图view()函数
原创 2019-08-05 20:41:35 · 53 阅读 · 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 · 8835 阅读 · 3 评论