reinitialize parameters in keras

最新推荐文章于 2022-11-24 18:52:52 发布

转载最新推荐文章于 2022-11-24 18:52:52 发布 · 394 阅读

python 同时被 2 个专栏收录

22 篇文章

订阅专栏

15 篇文章

订阅专栏

本文介绍了一种在Keras中重新初始化模型权重的方法。通过遍历模型的所有层及其变量，并利用初始设置重新运行初始化过程，可以有效地实现模型权重的重置。

# Reset/Reinitialize model weights/parameters · Issue #341 · keras-team/keras
# https://github.com/keras-team/keras/issues/341
session = K.get_session()
for layer in model.layers: 
     for v in layer.__dict__:
         v_arg = getattr(layer,v)
         if hasattr(v_arg,'initializer'):
             initializer_method = getattr(v_arg, 'initializer')
             initializer_method.run(session=session)
             print(f'reinitializing layer {layer.name}.{v}')