美文网首页
[PyTorch] 训练的时候很好测试的时候爆显存

[PyTorch] 训练的时候很好测试的时候爆显存

作者: VanJordan | 来源:发表于2018-12-15 17:54 被阅读11次
  1. 先设置不使用梯度,然后将测试时候的batchsize设置成训练时候的二分之一或者三分之一就不会爆了。
  2. 可能原因是测试的时候真的需要更大的显存。
def early_test(self):
    with torch.no_grad():
        self.model.eval()
        for path in tqdm(self.save_path,desc='Test Loop:'):
            self.eval(path)
        self.test()

相关文章

  • [PyTorch] 训练的时候很好测试的时候爆显存

    先设置不使用梯度,然后将测试时候的batchsize设置成训练时候的二分之一或者三分之一就不会爆了。 可能原因是测...

  • pytorch out of memory

    使用torch.cuda.empty_cache()删除一些不需要的变量代码示例如下: 测试的时候爆显存有可能是忘...

  • (16)验证集上内存溢出问题

    在训练过程中一直运行良好,但是在验证集上测试的时候内存溢出,GPU显存其实很大,batch size 也不是很大。...

  • Pytorch 节省显存的训练方法总结

    前言 最近的工作中,用到了Pytorch框架训练医学图像分割模型。精心设计的模型经常会因为显存不足而失败。减小模型...

  • pytorch显存管理

    PyTorch使用缓存分配器来加速内存分配。允许在不进行设备同步的情况下快速重新分配内存。缓存分配器中未被占用得内...

  • PyTorch 半精度训练踩坑

    背景 因为显卡显存不够,所以了解了一些PyTorch节省显存的方法:拿什么拯救我的 4G 显卡 - OpenMML...

  • Pytorch训练Resnet101,显存爆炸

    最近需要把caffe上的resnet101网络模型迁移到pytorch上,caffe上显存还够使,一到pytorc...

  • pytorch中碰到的memory leak问题

    最近碰到pytorch分布式训练时候,memory几乎线性增加,撑炸机器的问题。pytorch中内存泄漏常见的原因...

  • pytorch模型加载与保存

    pytorch模型训练流程 配置超参数: epoch,learning_rate 构建数据集:训练集,验证集,测试...

  • Pytorch Lenet Cifar10

    pytorch 实现Lenet 在cifar10数据集上的训练和测试 运行过程

网友评论

      本文标题:[PyTorch] 训练的时候很好测试的时候爆显存

      本文链接:https://www.haomeiwen.com/subject/dciahqtx.html