adding trainer solver property decorator

This commit is contained in:
Dario Coscia
2023-09-19 15:21:45 +02:00
committed by Nicola Demo
parent 1936133ad5
commit 939353f517

View File

@@ -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