pytorch
Cris_Lee卡卡卡
https://github.com/lrs1353281004
展开
-
pytorch踩坑记录-Error: Expected more than 1 value per channel when training
pytorch踩坑记录-Error: Expected more than 1 value per channel when training原创 2020-08-27 15:45:33 · 11335 阅读 · 10 评论 -
pytorch使用教程-基于自定义 Dataloader中的collate_fn 函数 实现变长数据处理
问题背景想要使用pytorch 框架中的 Dataset 和 Dataloader 类,将变长序列整合为batch数据 (主要是对长短不一的序列进行补齐),通过自定义collate_fn函数,实现对变长数据的处理。主要思路Dataset 主要负责读取单条数据,建立索引方式。Dataloader 负责将数据聚合为batch。应用实例测试环境: python 3.6 ,pytorch 1.2.0数据路径:data路径下存储的是待存储的数据样本。举例:其中的 1.json 样本格式为:定义原创 2020-05-14 22:13:02 · 8161 阅读 · 3 评论 -
pytorch、tensorflow-多gpu显存使用问题(解决gpu 0 显存占满时 使用其他gpu 报错 out of memory问题)
待补充https://github.com/pytorch/pytorch/issues/3477原创 2019-12-02 15:15:25 · 3328 阅读 · 0 评论