How to save the model?

[philipwelia]

I want to save models during training as the state_dict() function in pytorch. How can?

[AlirezaAzadbakht]

you can look at each ML or DL model as a set of parameters, if you completely know the model which you are working with, you can manually save those parameters, if not, a simple way to save ML models is using joblib library:

import joblib 
joblib.dump(model, 'model.pkl')

and simply load it:

model = joblib.load('model.pkl')

[philipwelia]

您好，我是黄绮恒，我已经收到了您的来信，非常感谢您能抽出宝贵时间给我发Email，我会尽快回复。祝您生活愉快！Hello, I am Huang Qiheng. I have received your letter. Thank you very much for taking the time to email me. I will reply as soon as possible.