Saving the model using joblib etc.

[lawrencepj13]

Hi there,

I'm trying to save the model to re deploy quickly using test data sets but have had issues using pickle/joblib etc.
Is there a method or built in function to save the model? or any recommendations on how to save it?

Many thanks.

[Anivader]

Hello,

I would recommend using this library to run DeepHit using PyTorch - https://github.com/havakv/pycox/blob/master/pycox/models/deephit.py

You can use the "save_model_weights" method to do this.