Skip to content

fix: Fix the problem where the model.save_weights() can't save multi-layer weight#1137

Merged
Oceania2018 merged 1 commit intoSciSharp:masterfrom
Beacontownfc:mybranch2
Jul 8, 2023
Merged

fix: Fix the problem where the model.save_weights() can't save multi-layer weight#1137
Oceania2018 merged 1 commit intoSciSharp:masterfrom
Beacontownfc:mybranch2

Conversation

@Beacontownfc
Copy link
Contributor

model.save_weights : When the weight name is nested multiple times (e.g. when the model has two weight names: bert/layer/dense/kernal:0 and bert/layer/dense_1/bias:0 the weight saved by bert/layer/dense/kernal:0 may overwritten by bert/layer/dense_1/bias:0), the API doesn't work as expected.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants