cair / tmu

Implements the Tsetlin Machine, Coalesced Tsetlin Machine, Convolutional Tsetlin Machine, Regression Tsetlin Machine, and Weighted Tsetlin Machine, with support for continuous features, drop clause, Type III Feedback, focused negative sampling, multi-task classifier, autoencoder, literal budget, and one-vs-one multi-class classifier. TMU is written in Python with wrappers for C and CUDA-based clause evaluation and updating.
https://pypi.org/project/tmu/
MIT License
129 stars 14 forks source link

Error when trying to pickle TM object #72

Open Mayur1009 opened 6 months ago

Mayur1009 commented 6 months ago

Hi, I am training a Tsetline Machine, and trying to save the trained model using pickle. When I train the TM using CPU, the model is saved without errors. But, if the model is trained on GPU, I get the following error:

Screenshot 2024-05-07 at 6 16 36 PM