pytorch
bob_chen_csdn
这个作者很懒,什么都没留下…
展开
-
pytorch GPU内存耗尽
pytorch GPU 计算过程中出现内存耗尽Pytorch GPU运算过程中会出现:“cuda runtime error(2): out of memory”这样的错误。通常,这种错误是由于在循环中使用全局变量当做累加器,且累加梯度信息的缘故,用官方的说法就是:"accumulate history across your training loop"。在默认情况下,开启梯度计算的Tenso...原创 2018-10-29 22:01:04 · 14670 阅读 · 0 评论 -
pytorch 存储模型
Pytorch 模型存储pytorch 模型存储有两种方法:存储参数#存储torch.save(the_model.state_dict(),PATH)#恢复the_model=THe_model()the_model.load_state_dict(torch.load(PATH))存储整个模型#存储torch.save(the_model,PATH)#恢复the_mo...原创 2018-10-29 22:51:46 · 444 阅读 · 0 评论 -
pytorch数据导入以及预处理
Pythorch 数据初始化pytorch在数据从原始数据集里面获取以后(一般处理成numpy数组),需要以下步骤:1.构造DataClasstorch.utils.data.Dataset是一个表示数据的抽象类,在构造自己的数据集时候,首先应该构造一个他的子类,并且该子类需要重写下面两个方法:__len__,这个方法在使用len(Dataset)时候会被调动,用于返回数据集Datase...原创 2018-11-01 17:32:57 · 6128 阅读 · 0 评论 -
pytorch动态网络以及权重共享
pytorch 动态网络+权值共享pytorch以动态图著称,下面以一个栗子来实现动态网络和权值共享技术:# -*- coding: utf-8 -*-import randomimport torchclass DynamicNet(torch.nn.Module): def __init__(self, D_in, H, D_out): """ ...原创 2018-11-02 20:29:27 · 7033 阅读 · 0 评论