Training config for the CUSTOM_TRAINED_MODEL problem type
KEY | TYPE | Description |
---|---|---|
MAX_DIMENSION | int | Maximum expected dimension of the catalog. |
INDEX_OUTPUT_PATH | str | Fully qualified cloud location (GCS, S3, etc) to export snapshots of the embedding to. |
DOCKER_IMAGE_URI | str | Docker image URI. |
SERVICE_PORT | int | Service port. |
MAX_CATALOG_SIZE | int | Maximum expected catalog size. |
STREAMING_EMBEDDINGS | bool | Flag to enable streaming embeddings. |