AttributeError: 'DeepHitSingle' object has no attribute 'state_dict'

[zachfreitas]

I'm trying to save my model using tensor flow save and I get the following Error when I use state_dict.
mdl.state_dict()
I get the follwing
AttributeError: 'DeepHitSingle' object has no attribute 'state_dict'

Is there a way around this?

[mahootiha-maryam]

Firstly pycox is based on pytorch, not tensorflow. Also for saving models you can not use state_dict you can save models in pycox in this way.

model.save_model_weights('myweights.pt')
model.load_model_weights('myweights.pt')

Also there was a previous issue on this question that havard explained it.