Skip to content

How to save the checkpoint for self-defined model? #265

@ziweiyang

Description

@ziweiyang

Hi,
I am interested about saving the checkpoint of my self-defined model by subclassing snt.Module, but I couldn't find solid resource online. Can I use default method in tensorflow 2 for keras model like tf.keras.callbacks.ModelCheckpoint()? Does this work for sonnet 2 model?

Thanks,
ZY

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions