在一个epoch中,sampler相当于把整个数据集
划分成了nproc_per_node份,
每个GPU每次得到batch_size的数量,
也就是nproc_per_node 个GPU分一整份数据集,
总数据量大小就为1个dataset
在一个epoch中,sampler相当于把整个数据集
划分成了nproc_per_node份,
每个GPU每次得到batch_size的数量,
也就是nproc_per_node 个GPU分一整份数据集,
总数据量大小就为1个dataset
举报文章
请选择举报类型
补充说明
0/200
上传截图
格式支持JPEG/PNG/JPG,图片不超过1.9M