def test_metrics(self):
return {
metric_name: to_tensor(test_metric.compute()).cpu()
for metric_name, test_metric in self._test_metrics.items()
}
def valid_metrics(self):
return {
metric_name: to_tensor(valid_metric.compute()).cpu()
for metric_name, valid_metric in self._valid_metrics.items()
}
def step_test_metrics(self):
return {
metric_name: to_tensor(step_test_metric).cpu()
for metric_name, step_test_metric in
self._step_test_metrics.items()
}
def test_loss(self):
return {
loss_name: to_tensor(train_loss.compute()).cpu()
for loss_name, train_loss in self._test_loss.items()
}
def step_test_loss(self):
return {
loss_name: to_tensor(step_test_loss).cpu()
for loss_name, step_test_loss in self._step_test_loss.items()
}
