sonnet icon indicating copy to clipboard operation
sonnet copied to clipboard

Published 20 hours ago • verified publisher icon google-deepmind

How to save the checkpoint for self-defined model?

Open ziweiyang opened this issue 1 year ago • 0 comments

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

ziweiyang avatar Apr 22 '23 00:04 ziweiyang