![](https://img-blog.csdnimg.cn/20201014180756913.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
文章平均质量分 74
dd_Zing
和光同尘,与君共勉
展开
-
配置(GPU/CPU)Pytorch环境以及修改conda和pip默认安装路径,安装环境前必看!!(———记自己装环境踩过的坑......)
在下载了Anaconda之后安装pytorch环境时,一般都是默认下载在C盘路径下,这里先要修改默认环境下载 路径,再下载pytorch,可以给C盘节省内存。保险起见,还可以设置Anaconda文件夹属性,权限全部勾选,其中的envs文件夹属性也可以如此设置。路径在Aconconda\envs中即为正确。打开Anaconda prompt。之后点击 ' y ' 确定安装。原创 2023-03-27 18:58:24 · 6816 阅读 · 4 评论 -
Pytorch笔记:
首先,在for循环中使用了DataLoader,进入DataLoader后,首先根据是否使用多进程DataLoaderIter,做出判断之后单线程还是多线程,接着使用Sampler得索引Index,然后将索引给到DatasetFetcher,在这里面调用Dataset,根据索引,通过getitem得到实际的数据和标签,得到一个batch size大小的数据后,通过collate_fn函数整理成一个Batch Data的形式输入到模型去训练。,也就说将样本都训练一遍,称为一个Epoch。原创 2023-03-26 16:21:01 · 130 阅读 · 0 评论