
PYTORCH
文章平均质量分 65
当年明月7
等风来~
展开
-
pytorch---Cosine learning rate scheduler
余弦学习率衰减公式:需要用到lambda 表达式:lambda:epoch: (epoch / 40) if epoch < 40else 0.5 * (math.cos(40/100) * math.pi) + 1)初始学习率为lr=10^-3如果是总的100个epoch,只使用后60个应用余弦学习率衰减,那么前40个不作cosine计算,前40个epoch的学习率为:lr1=epoch/40*lr, 那么后60个epoch适用0.5 * (math.cos(40/100).原创 2021-09-08 20:38:04 · 4476 阅读 · 0 评论 -
pytorch图像加载问题
使用pytorch制作图像数据集时,需要将存储在磁盘、硬盘的图像读取到内存中,涉及到图像I/O问题。在python中,图像处理主要采用的库:skimage, opencv-python, Pillow (PIL)。 这三个库均提供了图像读取的方法。三种主流图像处理库的比较:实验内容读取图像准备一张测试图像,彩色32bitimport matplotlib.pyplot as pltimport skimage.io as ioimport cv2from PIL impo原创 2021-07-27 15:18:50 · 970 阅读 · 0 评论 -
Linux下安装各种程序和遇到的问题解决方案汇总
1.pytorch安装安装完CUDA后,进入pytorch官网,https://pytorch.org/选取配置获取命令:pip3 install torch torchvision torchaudio网速非常差的,可以用镜像设置临时的Pip源命令如下:pip3 install torch torchvision torchaudio -i https://pypi.mirrors.ustc.edu.cn/simple/对是否安装成功进行测试:首先输入:python3...原创 2021-05-21 14:45:42 · 444 阅读 · 0 评论