- 博客(2)
- 收藏
- 关注
原创 pytorch DataLoader中的参数:collate_fn函数
def default_collate(batch): r"""Puts each data field into a tensor with outer dimension batch size""" error_msg = "batch must contain tensors, numbers, dicts or lists; found {}" elem_typ...
2019-06-30 19:50:14 2747
原创 pytorch DataLoader源码解析
#文章源码来自:pytorch源码。一、DataLoader的源码:class DataLoader(object): r""" Data loader. Combines a dataset and a sampler, and provides single- or multi-process iterators over the dataset. ...
2019-06-30 19:45:19 3447
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人