首页 > 其他分享 >torch.utils.data.dataloader中的collate_fn参数

torch.utils.data.dataloader中的collate_fn参数

时间:2022-10-06 21:34:26浏览次数:47  
标签:None torch utils dataloader collate data fn

torch.utils.data.DataLoader是pytorch提供的数据加载类,初始化函数如下:

torch.utils.data.DataLoader(dataset,batch_size=1, shuffle=False, 
sampler=None,batch_sampler=None, num_workers=0, 
collate_fn=<function default_collate>,pin_memory=False, 
drop_last=False, timeout=0, worker_init_fn=None)

  

  

   

标签:None,torch,utils,dataloader,collate,data,fn
From: https://www.cnblogs.com/zjuhaohaoxuexi/p/16758553.html

相关文章