nfflr.train.train

nfflr.train.train(model: Module, dataset: AtomsDataset, config: TrainingConfig, local_rank: int = 0)

NFFLr trainer entry point.

Parameters:

modeltorch.nn.Module

datasetnfflr.AtomsDataset

confignfflr.train.TrainingConfig

local_rankint, optional