Muti3d / train /multinode.env.example
xixibuxixi's picture
Upload Muti3d project code without data or checkpoints
1e073ab verified
Raw History Blame Contribute Delete
661 Bytes
# Copy to multinode.env
# Local node, rank 0.
HEAD_HOST=10.0.0.1
MASTER_ADDR=10.0.0.1
MASTER_PORT=29500
# Common training shape.
NNODES=2
NPROC=8
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
# Remote node, rank 1.
REMOTE_HOST=10.0.0.2
REMOTE_USER=root
REMOTE_TRAIN_DIR=/public/0_DATA/2_Avatar/zhizhou_share/rcz/Muti3d/train
# Prefer SSH keys. If password auth is needed, install sshpass on the head node
# and set REMOTE_PASSWORD below. Keep this file private.
REMOTE_PASSWORD=
# Optional shared overrides. Leave empty to use _run_stage2_eval_wandb_core.sh defaults.
RUN_NAME=muti3d_stage2_eval_wandb_multinode
DATA_ROOT=
INIT_DCP=
OUTPUT_ROOT=
WANDB_MODE=online