forward key metrics (#3467)

This commit is contained in:
William Falcon 2020-09-11 15:59:14 -04:00 committed by GitHub
parent d1d48e2ea1
commit 9e33a720e3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 0 deletions

View File

@ -43,6 +43,22 @@ class TrainerProperties(ABC):
def callback_metrics(self, x):
self.logger_connector.callback_metrics = x
@property
def logged_metrics(self):
return self.logger_connector.logged_metrics
@logged_metrics.setter
def logged_metrics(self, x):
self.logger_connector.logged_metrics = x
@property
def progress_bar_metrics(self):
return self.logger_connector.progress_bar_metrics
@progress_bar_metrics.setter
def progress_bar_metrics(self, x):
self.logger_connector.progress_bar_metrics = x
@property
def state(self) -> TrainerState:
return self._state