Training

Training#

本章主要讲解训练过程需要接触的概念与内容,主要包含:

  1. collate_function:在DataLoader里面操作BaseArrowDataset

  2. BaseSeqDataModule:在Lightning的体系里面,如果使用Lightning Trainer进行训练,需要用户编写LightningDataModuleTrajDL提供了封装好的BaseSeqDataModule

  3. LightningCLITrajDL支持使用LightningCLI进行模型训练与验证。