Save and reuse a trained model

[BBounajma]

Dear team and community,

Thank you for your work. I am currently trying to use DeepTab to implement a Saint-based classifier and I struggle with saving and loading my trained models. It seems that the save method defined in deeptab/base_models/utils/basemodel.py does not work for the SAINTclassifier class.
I also tried using joblib and manually saving the weights in a pth file, but in doing so the internal state (such as the task_model attribute) is lost. I therefore wanted to know how could I save my trained models locally.

Thank you in advance for your help !