AttributeError: 'DeepHitSingle' object has no attribute 'state_dict'
Open
- Dominant language
- Python
- Stars
- 995
- Forks
- 203
- PR merge metrics
- No merged PRs in 30d
Description
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?
Contributor guide
No contributing guide indexed for this repository
Research direction
Start with the DeepHitSingle instance and the mdl.state_dict() call, then compare that usage with the model's supported persistence approach. Done means identifying a way to save and restore DeepHitSingle without triggering the AttributeError.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 3/5
- Estimated time
- 1-2 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100