晨哥的技术分享后花园

摘要： 一开始写这篇随笔的时候还没有了解到 Dateloader有一个 collate_fn 的参数，通过定义一个collate_fn 函数，其实很多batch补齐到当前batch最长的操作可以放在collate_fn 里面去，这样代码在训练和模型中就可以更加简洁。有时间再整理一下这个吧。 ________ 阅读全文