pytroch
Yolo_C
奋斗在bug与debug之间
展开
-
【搬运】pytorch,keras,tensorflow的模型代码框架
pytorch,keras,tensorflow的模型代码框架 参考自此公众号文章,后续会不断地加详细一些 pytorch from torch.optim as Optimizer from torch import nn import torch class modelName(nn.Module): def init(self,): # 加入每个层的实例化 # 常用如下 nn.Embedding() nn.RNN()原创 2021-01-03 22:45:02 · 250 阅读 · 0 评论 -
pytorch并行报错RuntimeError: unable to write to file</torch_xxx>
问题主要是使用multiprocessing在每一轮训练时并行随机产生数据时,job.get()数据超过了torch的暂存内存,参考了以下文章,加入代码或者降低了num_pool都没有解决我的问题 https://blog.csdn.net/u012796629/article/details/105936386/ 最终我使用del data,在每轮训练结束时释放掉了该轮使用的数据就可以了 ...原创 2020-09-22 17:35:08 · 3986 阅读 · 1 评论 -
DEBUG:pytorch训练时候没有问题,预测时候内存迅速增长,爆掉,out of memory的解决办法
在使用pytorch训练模型的时候,训练时候,内存十分平稳,预测的时候迅速增长,报内存不足的错误 原始是忘记加no_grad 解决方法:在预测的时候,加with torch.no_grad() with torch.no_grad(): for i,batch in enumerate(test_loader): pass ...原创 2020-06-03 16:36:46 · 2822 阅读 · 3 评论 -
pytorch入门笔记
pytorch 张量入门与numpy的张量的关系 pytorch 张量与梯度 pytorch 神经网络(LeNet) pytorch 实现cifar10分类器 pytorch 模型保存与重载 https://github.com/yolochai/jupyternb/blob/master/pytorch_notebook.ipynb 顺便抹把泪,conda安装pytorch和torchvisio...原创 2020-01-30 02:24:31 · 233 阅读 · 1 评论 -
pytorch模型存取与加载
# 存取模型 PATH='./xxx.xx' # 模型名字 torch.save(net.state_dict(),PATH) # 保存,net为需要保存的网络 pretrained_net = torch.load(PATH) # 读取 net2 = Net() # Net()为保存的模型同结构的模型 net2.load_state_dict(pretrained_net) # 加载权重 ...原创 2020-01-29 00:16:49 · 545 阅读 · 0 评论