怎么冻结模型前面的层,只对后面的层进行Transfer Learning
bq3bfh9z1#
model的定义在这里: link ,你希望冻结哪个模块,把那里所有的参数trainable设置为False即可。可以参考这里进行设置。
1条答案
按热度按时间bq3bfh9z1#
model的定义在这里: link ,你希望冻结哪个模块,把那里所有的参数trainable设置为False即可。可以参考这里进行设置。