Skip to content

Instantly share code, notes, and snippets.

@sherwoac

sherwoac/removes_groups_from_model_config.py

Created August 5, 2024 17:17
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save sherwoac/180438367d4ca009e3672632247dac30 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save sherwoac/180438367d4ca009e3672632247dac30 to your computer and use it in GitHub Desktop.
Download ZIP
scripts to change model config from keras 2->3 compliant
Raw
removes_groups_from_model_config.py
# hack to change model config from keras 2->3 compliant
import h5py
f = h5py.File(savedModelPath, mode="r+")
model_config_string = f.attrs.get("model_config")
if model_config_string.find('"groups": 1,') != -1:
model_config_string = model_config_string.replace('"groups": 1,', '')
f.attrs.modify('model_config', model_config_string)
f.flush()
model_config_string = f.attrs.get("model_config")
assert model_config_string.find('"groups": 1,') == -1
f.close()
@ptishkin
Copy link

ptishkin commented Jun 22, 2025 

    training_config_string = f.attrs.get("training_config")
    if training_config_string.find('"loss": "mae"') != -1:
        training_config_string = training_config_string.replace('"loss": "mae"', '"loss": "mean_absolute_error"')
        f.attrs.modify('training_config', training_config_string)
        f.flush()
        training_config_string = f.attrs.get("training_config")
        assert training_config_string.find('"loss": "mae"') == -1

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