一开始我还在嘲笑,为啥作者的代码要利用这个shape信息来动态地获取batch_size。
回答很简单,emmm,因为在最后一个batch中,数据可能不足一个设定的batch_size。
【Debug】 运行时的 batch_size 跟config中的 batch_size 不是一个东西。
阅读 129
2023-02-13
一开始我还在嘲笑,为啥作者的代码要利用这个shape信息来动态地获取batch_size。
回答很简单,emmm,因为在最后一个batch中,数据可能不足一个设定的batch_size。
相关推荐
精彩评论(0)