Skip to content

nmdl-mizo/gnn_train_container

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

79 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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>