pytorch
关注
分享
扫一扫
文章平均质量分 92
yuanzhoulvpi
喜欢写代码的数据分析师
展开
-
python tips003 ——DataLoader的collate_fn参数使用详解
背景最近在看sentences-transformers的源码，在有一个模块发现了dataloader.collate_fn，当时没搞懂是什么意思，后来查了一下，感觉还是很有意思的，因此来分享一下。dataloaderdataloader肯定都是知道的，就是为数据提供一个迭代器。基本工作机制：在dataloader按照batch进行取数据的时候, 是取出大小等同于batch size的index列表，然后将列表列表中的index输入到dataset的getitem()函数中，取出该index对应的原创 2022-01-03 10:38:10 · 5059 阅读 · 0 评论 -
RuntimeError: cuda runtime error (30) ...work/aten/src/THC/THCGeneral.cpp:51
我的电脑是ubuntu18.10，python是3.7.1(miniconda)。我中午睡午觉，电脑熄屏，然后pytorch 的cuda就不能用了，包括tensorflow都不行，连nvidia的sample都不能运行了，原创 2019-08-18 16:56:36 · 1094 阅读 · 1 评论