Implementing fix from Issue#344

[saskiabosma]

Hello everyone, I had saving issues on YoutubeDNN, and the fix mentioned by user cbjs in the issue about this error 344 solved it. This fix is not implemented in theDeepCTR library, is there anything blocking it ? If not, I'd be happy to contribute an MR !

Additional context
There is a recent comment on an old issue which I think is related: Issue43.

Operating environment(运行环境):

• python version [3.9]
• tensorflow version [2.13.0]
• deepctr version [0.9.3]