test_metrics.json

{'test_loss': 0.5517849326133728, 'test_r2': 0.64, 'test_spearmanr': 0.7761873360094849, 'test_runtime': 25.1806, 'test_samples_per_second': 301.621, 'test_steps_per_second': 18.864}