PicklingError while trying to save the model

[SaraAlthubaiti]

Hi all,

I used CoxPH and at the end when trying to save the trained model with save_net like that model.save_net('data_root/model'), I get this error:

PicklingError: Can't pickle <class 'torch._C._VariableFunctions'>: it's not the same object as torch._C._VariableFunctions

Any help with that, please?

[havakv]

Hi! I'm not sure what this error might come from. For debugging could you provide the following details?

• pycox version, you can get it with pycox.__version__
• How you installed pycox (with pip or from source?)
• pytorch version (torch.__version__)
• OS (mac, linux, windows)
• An example resulting in the error (I need to know how you specify your net, your model, etc). Preferably, I should be able to run your code to reproduce the error.
• The full error trace (it should tell you exactly which line in the code failed)