Training#
本章主要讲解训练过程需要接触的概念与内容,主要包含:
collate_function:在DataLoader里面操作BaseArrowDataset。BaseSeqDataModule:在Lightning的体系里面,如果使用Lightning Trainer进行训练,需要用户编写LightningDataModule,TrajDL提供了封装好的BaseSeqDataModule。LightningCLI:TrajDL支持使用LightningCLI进行模型训练与验证。