nfflr.train.trainer.get_dataflow

nfflr.train.trainer.get_dataflow(dataset: nfflr.AtomsDataset, config: TrainingConfig)

Configure training and validation datasets.

Wraps train and val splits of dataset in ignite’s auto_dataloader.

Parameters:

datasetnfflr.AtomsDataset

confignfflr.train.TrainingConfig