timeseriesAI / tsai

Time series Timeseries Deep Learning Machine Learning Python Pytorch fastai | State-of-the-art Deep Learning library for Time Series and Sequences in Pytorch / fastai
https://timeseriesai.github.io/tsai/
Apache License 2.0
5.14k stars 642 forks source link

TSClassifier issue #886

Open ysmin-g opened 7 months ago

ysmin-g commented 7 months ago

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'