test_metrics.json

{'test_loss': 0.6263986229896545, 'test_r2': 0.768, 'test_spearmanr': 0.8726603041029465, 'test_runtime': 44.7294, 'test_samples_per_second': 159.939, 'test_steps_per_second': 10.016}