batch_size=loader_batch
时间: 2023-11-24 19:06:12 浏览: 84
Lenet_patternrecognition_goldz96_zip_
这段代码中的 `batch_size=loader_batch` 是为了设置每个 batch 的样本数。其中,`args.batch_size` 表示用户从命令行传入的 batch size 参数,`args.num_pos` 表示每个样本对应的正样本数量。因此,`loader_batch` 表示一个 batch 中实际上包含的样本数,即 `batch_size * num_pos`。这个值通常会比 `batch_size` 大很多,因为在训练 triplet loss 或者 contrastive loss 时,每个样本都需要与其它样本进行比较,生成正样本和负样本,因此需要使用更多的样本。
阅读全文