![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
仙草冻奶茶
这个作者很懒,什么都没留下…
展开
-
Pytorch:torch.optim
torch.optim文档 构建 要使用torch.optim,必须构造一个optimizer对象。这个对象能保存当前的参数状态并且基于计算梯度更新参数。 构造optimizer必须给它一个包含参数(必须都是Variable对象)进行优化。然后,可以指定optimizer的参数选项,比如学习率,权重衰减等。 如下: optimizer = optim.SGD(model.parameters(),...原创 2019-10-16 14:45:53 · 190 阅读 · 0 评论 -
Pytorch:torch.utils.data.Dataset类和torch.utils.data.DataLoader类
torch.utils.data.Dataset类 Dataset是一个包装类,用来将数据包装为Dataset类,然后传入DataLoader中。他是一个重写类,可以通过继承和重写这个类就可以定义自己的数据类,需要重定义__len__(),和__getitem__()这两个函数。 len():返回数据的长度 getitem():获取数据的方式 torch.utils.data.DataLoad...原创 2019-10-14 16:53:02 · 760 阅读 · 0 评论