Hi slightly smiling face I have testing Temporal Fusion Tran

Question

Hi slightly smiling face I have testing Temporal Fusion Transformer in the package but I tend to run out of cuda memory ```OutOfMemoryError CUDA out of memory ``` Is there a parameter to limit that or any recommendations on how to avoid that

Cristian (Nixtla) · Answer

Hi < Tuhin Mallick> In our experience TFT is highly memory intensive You can limit the memory usage with `windows batch size` and `hidden size` parameters

Kin Gtz. Olivares · Answer

You can also try to reduce the size of the autorregresive features through the `input size`

Kin Gtz. Olivares · Answer

Hey < Tuhin Mallick> We fixed this week TFT s memory usage We will merge that PR into main soon If you want to try the changes you can install them with this line ``` pip install git+```