[Minor] Improve performance based on profiling results

[MaiBe-ctrl]

:microscope: Background

• Why is this change needed? Is there a related issue or a new feature to be added?

:crystal_ball: Key changes

• Explain the main changes introduced by this pull request for the reviewer.

:clipboard: Review Checklist

• [ ] I have performed a self-review of my own code.
• [ ] I have commented my code, added docstrings and data types to function definitions.
• [ ] I have added pytests to check whether my feature / fix works.

Please make sure to follow our best practices in the Contributing guidelines.

[github-actions[bot]]

Model Benchmark

Benchmark	Metric	main	current	diff
PeytonManning	MAE_val	0.35539	0.35539	0.0%	:white_check_mark:
PeytonManning	RMSE_val	0.50401	0.50401	0.0%	:white_check_mark:
PeytonManning	Loss_val	0.01803	0.01803	0.0%	:white_check_mark:
PeytonManning	train_loss	0.01466	0.01466	0.0%	:white_check_mark:
PeytonManning	reg_loss	0	0	0.0%	:white_check_mark:
PeytonManning	MAE	0.34755	0.34755	0.0%	:white_check_mark:
PeytonManning	RMSE	0.49449	0.49449	0.0%	:white_check_mark:
PeytonManning	Loss	0.01465	0.01465	0.0%	:white_check_mark:
PeytonManning	time	55.9486	58.87	5.22%	:warning:
YosemiteTemps	MAE_val	0.59734	0.59734	0.0%	:white_check_mark:
YosemiteTemps	RMSE_val	0.88884	0.88884	0.0%	:white_check_mark:
YosemiteTemps	Loss_val	0.00046	0.00046	0.0%	:white_check_mark:
YosemiteTemps	train_loss	0.00126	0.00126	0.0%	:white_check_mark:
YosemiteTemps	reg_loss	0	0	0.0%	:white_check_mark:
YosemiteTemps	MAE	0.97461	0.97461	0.0%	:white_check_mark:
YosemiteTemps	RMSE	1.70806	1.70806	0.0%	:white_check_mark:
YosemiteTemps	Loss	0.00126	0.00126	0.0%	:white_check_mark:
YosemiteTemps	time	210.194	207.69	-1.19%	:white_check_mark:
EnergyPriceDaily	MAE_val	5.64186	5.64186	0.0%	:white_check_mark:
EnergyPriceDaily	RMSE_val	7.19914	7.19914	0.0%	:white_check_mark:
EnergyPriceDaily	Loss_val	0.02893	0.02893	0.0%	:white_check_mark:
EnergyPriceDaily	train_loss	0.02957	0.02957	0.0%	:white_check_mark:
EnergyPriceDaily	reg_loss	0	0	0.0%	:white_check_mark:
EnergyPriceDaily	MAE	6.39655	6.39655	0.0%	:white_check_mark:
EnergyPriceDaily	RMSE	8.56207	8.56207	0.0%	:white_check_mark:
EnergyPriceDaily	Loss	0.02936	0.02936	0.0%	:white_check_mark:
EnergyPriceDaily	time	58.1817	62.45	7.34%	:x:
AirPassengers	MAE_val	30.8328	30.8328	0.0%	:white_check_mark:
AirPassengers	RMSE_val	31.8182	31.8182	0.0%	:white_check_mark:
AirPassengers	Loss_val	0.01301	0.01301	0.0%	:white_check_mark:
AirPassengers	train_loss	0.00071	0.00071	0.0%	:white_check_mark:
AirPassengers	reg_loss	0	0	0.0%	:white_check_mark:
AirPassengers	MAE	6.88174	6.88174	0.0%	:white_check_mark:
AirPassengers	RMSE	8.92109	8.92109	0.0%	:white_check_mark:
AirPassengers	Loss	0.00076	0.00076	0.0%	:white_check_mark:
AirPassengers	time	12.689	13.71	8.05%	:x:

Model training plots

## Model Training ### PeytonManning  ### YosemiteTemps  ### AirPassengers  ### EnergyPriceDaily