Can ROCKET on the GPU be used with TSRegression()?

[jrfackler]

I've followed the tutorial 02_ROCKET_a_new_SOTA_classifier to use ROCKET on GPU but that tutorial is for a classifier like RidgeClassifierCV.

Is there a way to use ROCKET on the GPU with regression? I've try to use TSRegression() both with and without a loss function and I can't get it to work.

X, y, splits = combine_split_data([X_train, X_valid], [y_train, y_valid])

tfms = [None, TSRegression()]
batch_tfms = TSStandardize(by_sample=True)
dls = get_ts_dls(X, y, splits=splits, tfms=tfms, drop_last=False, shuffle_train=False, batch_tfms=batch_tfms, bs=10_000)
model = build_ts_model(ROCKET, dls=dls)
learn = Learner(dls, model, metrics=accuracy, cbs=ShowGraph())

X.shape, y.shape
((2125, 1525, 30), (2125,))

learn.fit_one_cycle(10, 3e-4)

Traceback (most recent call last):

  Cell In[3], line 1
    learn.fit_one_cycle(10, 3e-4)

  File ~\anaconda3\envs\quant\lib\site-packages\fastai\callback\schedule.py:114 in fit_one_cycle
    if self.opt is None: self.create_opt()

  File ~\anaconda3\envs\quant\lib\site-packages\fastai\learner.py:188 in create_opt
    self.opt = self.opt_func(self.splitter(self.model), lr=self.lr)

  File ~\anaconda3\envs\quant\lib\site-packages\fastai\optimizer.py:242 in Adam
    return Optimizer(params, cbs, lr=lr, mom=mom, sqr_mom=sqr_mom, eps=eps, wd=wd)

  File ~\anaconda3\envs\quant\lib\site-packages\fastai\optimizer.py:98 in __init__
    self.param_lists = L(L(p) for p in params) if isinstance(params[0], (L,list)) else L([params])

  File ~\anaconda3\envs\quant\lib\site-packages\fastcore\foundation.py:112 in __getitem__
    def __getitem__(self, idx): return self._get(idx) if is_indexer(idx) else L(self._get(idx), use_list=None)

  File ~\anaconda3\envs\quant\lib\site-packages\fastcore\foundation.py:116 in _get
    if is_indexer(i) or isinstance(i,slice): return getattr(self.items,'iloc',self.items)[i]

IndexError: list index out of range

[jrfackler]

I reinstalled tsai with the latest version 0.3.9 and that resolved the issue