如何保存自定义模型的检查点?
作者: ziweiyang创建于 2023年4月22日更新于 2023年4月22日
你好, 我对通过子类化 `snt.Module` 来保存自定义模型的检查点很感兴趣, 但我在网上找不到相关的资料。我是否可以使用 tensorflow 2 中的默认方法来为 keras 模型创建检查点, 比如 `tf.keras.callbacks.ModelCheckpoint()`? 这种方法是否适用于 sonnet 2 模型?
内容来源: google-deepmind/sonnet