nfflr.train.trainer.setup_model_and_optimizer#
- nfflr.train.trainer.setup_model_and_optimizer(model: torch.nn.Module, dataset: nfflr.AtomsDataset, config: TrainingConfig)[source]#
Initialize model, criterion, and optimizer.
Initialize model, criterion, and optimizer.