-
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>
-
run the trainer container
make train MODEL=<model name> ARG=<path to argument file>
-
Notifications
You must be signed in to change notification settings - Fork 0
nmdl-mizo/gnn_train_container
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published