Skip to content

Instantly share code, notes, and snippets.

@stsievert
Last active August 17, 2018 15:15
Show Gist options
  • Save stsievert/7389acd62eb6f6d151286214cb51c392 to your computer and use it in GitHub Desktop.
Save stsievert/7389acd62eb6f6d151286214cb51c392 to your computer and use it in GitHub Desktop.
Display the source blob
Display the rendered blob
Raw
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Display the source blob
Display the rendered blob
Raw
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@stsievert
Copy link
Author

stsievert commented Aug 15, 2018

I think this gist is with modifying the decorator at https://github.com/dask/distributed/blob/4e36660e84b0eb20c1bc4127ba07a1876aefd5c5/distributed/protocol/keras.py#L8 to be

@dask_serialize.register(
    (
        keras.Model,
        keras.engine.training.Model,
        keras.models.Model,
        keras.model.Sequential

    )
)

and likewise for serialization.

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