full list of usable variables in arch_config

[AslantheAslan]

I have been using tsai for a while now, but I still don't know all variables that we can pass to arch_config. For example, how can I define an LSTM structure that has 2 layers, where the first layer's size is [128,128] and the second layer's size is [128,2]?

[oguiza]

Hi @AslantheAslan,
There's no list of arguments that can be passed to models in tsai as each one of them allows different ones.
I'd recommend you take a look at the documentation for each architecture to understand what can be used.
Here's the link for LSTMPlus (it's an enhanced version of LSTM that allows more arguments).

[AslantheAslan]

Thanks for the response. I have overlooked LSTMPlus until now, I will definitely utilize its different arguments.