myogestic.models.definitions.raulnet_models.train

myogestic.models.definitions.raulnet_models.train(model, dataset, _, __)

Train a RaulNet model.

Parameters:

• model (L.LightningModule) – The RaulNet model to train.

• dataset (dict) – The dataset to train the model with.

• _ (bool) – If the model is a classifier.

• __ (CustomLogger) – The logger to log the training process.

Returns:

The trained RaulNet model.

Return type:

L.LightningModule