TSClassifier issue

[ysmin-g]

I am using the TSClassifier to implement the TST model for time series classification. Every time I run the code, I have been receiving the error below:

/usr/local/lib/python3.10/dist-packages/tsai/models/TST.py in init(self, c_in, c_out, seq_len, max_seq_len, n_layers, d_model, n_heads, d_k, d_v, d_ff, dropout, act, fc_dropout, y_range, verbose, **kwargs)
172 self.new_q_len = True
173 t = torch.rand(1, 1, seq_len)
--> 174 q_len = nn.Conv1d(1, 1, **kwargs)(t).shape[-1]
175 self.W_P = nn.Conv1d(c_in, d_model, **kwargs) # Eq 2
176 pv(f'Conv1d with kwargs={kwargs} applied to input to create input encodings\n', verbose)

TypeError: Conv1d.init() got an unexpected keyword argument 'custom_head'