mirror of
https://github.com/twitter/the-algorithm-ml.git
synced 2024-11-18 22:19:23 +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"
|