Welcome to OStack Knowledge Sharing Community for programmer and developer-Open, Learning and Share
Welcome To Ask or Share your Answers For Others

Categories

0 votes
392 views
in Technique[技术] by (71.8m points)

tensorflow - Customize trainable parameters of KERAS model

Multi hidden units

I just want to set all the parameters to not trainable expect the ones that are connected to dense_5 layer which are it's own parameters and the parameters of dense_6 layer that are connected to dense_5. How can I set these last parameters to trainable ? By assigning the trainable to true for the dense_6 layer, all the parameters become trainable even if they are not connected to dense_5.

question from:https://stackoverflow.com/questions/66056589/customize-trainable-parameters-of-keras-model

与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome To Ask or Share your Answers For Others

1 Answer

0 votes
by (71.8m points)

You can not change trainable for a part of layer. Change the architecture: don't concat dense_5, add dense_7 after dense_5, concat outputs of dense_6 and dense_7. Then set trainable for dense_5 and Dense_7.


与恶龙缠斗过久,自身亦成为恶龙;凝视深渊过久,深渊将回以凝视…
Welcome to OStack Knowledge Sharing Community for programmer and developer-Open, Learning and Share
Click Here to Ask a Question

...