rvandewater / YAIB

🧪Yet Another ICU Benchmark: a holistic framework for the standardization of clinical prediction model experiments. Provide custom datasets, cohorts, prediction tasks, endpoints, preprocessing, and models. Paper: https://arxiv.org/abs/2306.05109
https://github.com/rvandewater/YAIB/wiki
MIT License
50 stars 9 forks source link

Memory optimisation for Temporal Fusion Transformer #146

Open youssefmecky96 opened 4 months ago

youssefmecky96 commented 4 months ago

It seems that the the TFT model is running out of memory when preforming hyper parameter tuning need to check if there is a leak somewhere using profiling and comparing how much memory do base models use. Then deciding how to proceed from there.

Out of memory