pytorch
_rookie_coder
Talk is cheap.Show me the code!
展开
-
linux下使用anaconda安装pytorch
pytorch本文主要介绍一下使用anaconda在在linux环境下的安装pytorch的流程。由于pytorch安装比较简单,这里只写出一些主要的流程。anaconda的安装在此就不再赘述。1.确定自己的系统版本因为pytorch在0.4已经不支持比较老的系统版本了。以下是pytorch的member在github上的原话。In 0.4, we deprecated R...原创 2018-06-12 16:27:33 · 25226 阅读 · 4 评论 -
pytorch学习笔记(1)-optimizer.step()和scheduler.step()
optimizer.step()和scheduler.step()的区别optimizer.step()通常用在每个mini-batch之中,而scheduler.step()通常用在epoch里面,但是不绝对,可以根据具体的需求来做。只有用了optimizer.step(),模型才会更新,而scheduler.step()是对lr进行调整。通常我们有optimizer = optim.SGD...原创 2018-10-18 18:24:17 · 90208 阅读 · 5 评论 -
pytorch学习笔记(2)-使用自定义txt文件读取数据
import osimport torchfrom PIL import Imagefrom torch.utils.data import Datasetfrom torchvision import datasets, models, transformsclass dataLoader(Dataset): def __init__(self, path, listName,...原创 2018-10-19 17:11:59 · 6455 阅读 · 1 评论