diff --git a/pina/trainer.py b/pina/trainer.py index 3250614..be220d2 100644 --- a/pina/trainer.py +++ b/pina/trainer.py @@ -36,3 +36,9 @@ class Trainer(pl.Trainer): def train(self, **kwargs): # TODO add kwargs and lightining capabilities return super().fit(self._model, self._loader, **kwargs) + @property + def solver(self): + """ + Returning trainer solver. + """ + return self._model \ No newline at end of file