[Minor] Simplify training procedure - flatten function calls #1642

ModelReport

Benchmark	Metric	main	current	diff
YosemiteTemps	MAE_val	0.58957	0.58957	0.0%	✅
YosemiteTemps	RMSE_val	0.86595	0.86595	0.0%	✅
YosemiteTemps	Loss_val	0.00044	0.00044	0.0%	✅
YosemiteTemps	MAE	0.9641	0.9641	0.0%	✅
YosemiteTemps	RMSE	1.69444	1.69444	0.0%	✅
YosemiteTemps	Loss	0.00123	0.00123	0.0%	✅
YosemiteTemps	LR	8e-05	8e-05	0.0%	✅
YosemiteTemps	time	139.394	141.13	1.25%	✅
EnergyPriceDaily	MAE_val	5.39142	5.39142	0.0%	✅
EnergyPriceDaily	RMSE_val	6.65739	6.65739	0.0%	✅
EnergyPriceDaily	Loss_val	0.02483	0.02483	0.0%	✅
EnergyPriceDaily	MAE	5.93446	5.93446	0.0%	✅
EnergyPriceDaily	RMSE	7.96693	7.96693	0.0%	✅
EnergyPriceDaily	Loss	0.02572	0.02572	0.0%	✅
EnergyPriceDaily	LR	0.00019	0.00019	0.0%	✅
EnergyPriceDaily	time	39.3679	38.94	-1.09%	✅
AirPassengers	MAE_val	30.8849	30.8849	0.0%	✅
AirPassengers	RMSE_val	32.0342	32.0342	0.0%	✅
AirPassengers	Loss_val	0.01319	0.01319	0.0%	✅
AirPassengers	MAE	6.49056	6.49056	0.0%	✅
AirPassengers	RMSE	8.45795	8.45795	0.0%	✅
AirPassengers	Loss	0.00071	0.00071	0.0%	✅
AirPassengers	LR	0.00022	0.00022	0.0%	✅
AirPassengers	time	9.55878	9.4	-1.66%	✅
PeytonManning	MAE_val	0.35144	0.35144	0.0%	✅
PeytonManning	RMSE_val	0.50166	0.50166	0.0%	✅
PeytonManning	Loss_val	0.01781	0.01781	0.0%	✅
PeytonManning	MAE	0.34701	0.34701	0.0%	✅
PeytonManning	RMSE	0.4942	0.4942	0.0%	✅
PeytonManning	Loss	0.01469	0.01469	0.0%	✅
PeytonManning	LR	0.00028	0.00028	0.0%	✅
PeytonManning	time	24.7513	25.48	2.94%	✅