Function withSaveHandler

  • withSaveHandler(
        saveHandler: (artifacts: ModelArtifacts) => Promise<SaveResult>,
    ): IOHandler

  • Creates an IOHandler that passes saved model artifacts to a callback.

    function handleSave(artifacts) {
      // ... do something with the artifacts ...
      return {modelArtifactsInfo: {...}, ...};
    }

    const saveResult = model.save(tf.io.withSaveHandler(handleSave));

    Parameters

    • saveHandler: (artifacts: ModelArtifacts) => Promise<SaveResult>

      A function that accepts a ModelArtifacts and returns a promise that resolves to a SaveResult.

    Returns IOHandler

Settings

Member Visibility