Cannot save learner in pickle format for Time series classification with Transformer (TST) model

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

4.91k stars 622 forks source link

Cannot save learner in pickle format for Time series classification with Transformer (TST) model #833

Open TanjinToma opened 10 months ago

TanjinToma commented 10 months ago

learn.save_all(path='export', dls_fname='dls', model_fname='model', learner_fname='learner') The above command shows the following error for the TST model TypeError: cannot pickle 'code' object

angiebao commented 3 months ago

I have exactly the same problem when using learn.export learn.export(f"./models/{learner_fname}.pkl")