mirror of
https://github.com/twitter/the-algorithm-ml.git
synced 2024-12-23 23:01:48 +01:00
10 lines
226 B
Bash
10 lines
226 B
Bash
#! /bin/sh
|
|
|
|
torchrun \
|
|
--standalone \
|
|
--nnodes 1 \
|
|
--nproc_per_node 2 \
|
|
/usr/src/app/tml/projects/twhin/run.py \
|
|
--config_yaml_path="/usr/src/app/tml/projects/twhin/config/local.yaml" \
|
|
--save_dir="/some/save/dir"
|