![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
依旧seven
这个作者很懒,什么都没留下…
展开
-
pytorch中loss为nan值
遇到的情况:在本地pytorch1.3.1运行无误。但是在服务器上,环境是pytorch1.2.0 则出现:loss变为NAN值,尝试过的方法:1)改变batchsize无作用。 2)调小学习率没作用。 3)数据做了归一化也没有作用 4)冻结了一些卷积层,也毫无作用但是加了一句话: preds = preds.log_softmax(2).detach().requires...原创 2020-01-03 17:00:01 · 1660 阅读 · 2 评论 -
pytorch中保存网络模型的两种方式
一、只保存网络中的参数保存:torch.save(model.state_dict(), save_fp)加载的时候需要先初始化一个模型,然后把文件中的参数恢复。train_weights = torch.load(model_fp)model = Model()model.load_state_dict(model_weights)这里load得到的是变量类型为Ord...原创 2019-12-26 13:53:11 · 963 阅读 · 2 评论 -
构建第一个网络——回归问题
记录下搭建的第一个神经网络,终于感觉到摸到了门槛。随机构建了一个y = x^2 的函数和一些扰乱点,预测y的值。 from torch.autograd import Variableimport torch.nn.functional as Fimport matplotlib.pyplot as pltx = torch.unsqueeze(torch.linspa...原创 2018-10-15 20:58:48 · 142 阅读 · 0 评论 -
torch.utils.data.DataLoader函数详解
class DataLoader(object): r""" Data loader. Combines a dataset and a sampler, and provides single- or multi-process iterators over the dataset. Arguments: dataset (Dataset): ...原创 2018-10-17 10:51:14 · 13713 阅读 · 5 评论 -
Pytorch建立卷积神经网络进行图片分类
def __init__(self, in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True): kernel_size = _pair(kernel_size) stride = _pair(stride) pad...原创 2019-06-25 21:49:20 · 2130 阅读 · 0 评论