
Pytorch
文章平均质量分 79
Jamence
AI从业者,负责过多模态大模型、超大规模分类、聚类、检索等任务;发表过10+ CCF A/B等论文
展开
-
torch.utils.data.dataset 的数据组织形式——python list、dict、tuple内存消耗量
在Pytorch中,我们需要通过来实现数据的读取。是一种非流式的数据读取策略,需要将数据一次性导入至内存中.如果数据规模过大,可能存在内存不够的问题。例如代码中,将所有image_path和label都写到self.data中,可能会爆内存。如何优化呢?原创 2024-11-19 09:28:03 · 747 阅读 · 0 评论 -
Focal Loss for Dense Object Detection(RetinaNet)论文及代码解析
@[TOC](Focal Loss for Dense Object Detection(RetinaNet)论文及代码解析)Focal Loss for Dense Object Detection(RetinaNet)论文及代码解析本文针对Focal Loss for Dense Object Detection做出自己的判断和解析,不一定代表正确性(鄙人并不做目标检测。。。)论文下载链接(arxiv国内镜像):http://xxx.itp.ac.cn/pdf/1708.02002v2论文代码原创 2020-05-18 00:23:39 · 608 阅读 · 1 评论 -
Pytorch Tutorial Autograd翻译
Pytorch Tutorial AutoGrad翻译转载 2019-03-02 18:09:03 · 308 阅读 · 0 评论