https://lacucinadiadine.blogspot.com/2024/02/gpu-oom-hyperparameter-tuning-loop-with.html