接着我们自定义一个collate_fn,这里可以使用任何名字,只要在DataLoader里面传入就可以了。 def collate_fn(batch): batch.sort(key=lambda x: len( ...
確定! 回上一頁