Skip to content

Instantly share code, notes, and snippets.

@elijahc

elijahc/keras_reset_weights.py

Created April 13, 2018 18:27
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save elijahc/eabbe365b3b2ba468d33be97e788fec8 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save elijahc/eabbe365b3b2ba468d33be97e788fec8 to your computer and use it in GitHub Desktop.
Download ZIP
Reset weights in a Keras layer
Raw
keras_reset_weights.py
import keras.backend as K
def reset_weights(model):
session = K.get_session()
for layer in model.layers:
if hasattr(layer, 'kernel_initializer'):
layer.kernel.initializer.run(session=session)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment