Open lawrencepj13 opened 1 year ago
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.
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.
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.