dgs.models.engine.dgs_engine.DGSEngine.train_model

DGSEngine.train_model() → torch.optim.Optimizer

Train the given model using the given loss function, optimizer, and learning-rate schedulers.

After every epoch, the current model is tested and the current model is saved.

Returns:

The current optimizer after training.