Skip to content

Model Registry

Repository Latest Release

Warning

This module is only available as a prerelease. It is not yet fully documented nor stable.

Cogment Model Registry is an out-of-the-box component. It is a versioned key value store dedicated to the storage of AI models used in Cogment actors.

Usage

Warning

🚧 under construction

Setup

Cogment Model Registry is available as a docker container on dockerhub as cogment/model-registry.

It can be launched using the following;

docker run -p 9000:9000 -v $(pwd)/relative/path/to/model/archive:/data cogment/model-registry

If you want to provide specific configuration as environment variables use docker -e option.

docker run -p 9000:9000 -e COGMENT_MODEL_REGISTRY_GRPC_REFLECTION=1 -v $(pwd)/relative/path/to/model/archive:/data cogment/model-registry

The following environment variables can be used to configure the server:

  • COGMENT_MODEL_REGISTRY_PORT: The port to listen on. Defaults to 9000.
  • COGMENT_MODEL_REGISTRY_ARCHIVE_DIR: The directory to store model archives. Docker images default to /data.
  • COGMENT_MODEL_REGISTRY_SENT_MODEL_VERSION_DATA_CHUNK_SIZE: The size of the model version data chunk sent by the server. Defaults to 510241024 (5MB).
  • COGMENT_MODEL_REGISTRY_GRPC_REFLECTION: Set to start a gRPC reflection server. Defaults to false.