Skip to content

Latest commit

 

History

History
23 lines (15 loc) · 471 Bytes

README.md

File metadata and controls

23 lines (15 loc) · 471 Bytes

GNN Train Containers

How to use

  1. build the image

    make build MODEL=<model name>

    Model name is the name of the context directory in ctx/MODEL.

    When installing packages from a private repository

    make build-git MODEL=<model name> GIT_USER=<user name> GIT_TOKEN=<token> VERSION=<version>

  2. run the trainer container

    make train MODEL=<model name> ARG=<path to argument file>