nfflr.train.lr

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

NFFLr learning rate finder entry point.

Runs the Fast.ai learning rate finder ignite.handlers.lr_finder.FastaiLRFinder for model, dataset, and config.

Parameters:

modeltorch.nn.Module

datasetnfflr.AtomsDataset

confignfflr.train.TrainingConfig

local_rankint, optional