batch_idx / len(train_loader)

由于读取数据时就按照BATCH_SIZE读取了,因此train_loader就是按照BATCH_SIZE分份的,每一份就是BATCH_SIZE大小,一共总数/BATCH_SIZE份。 batch_idx就是第几份。

batch_idx / len(train_loader) 就是已经训练的份数除以数据总份数。

猜你喜欢

转载自blog.csdn.net/weixin_42133481/article/details/114291501
今日推荐