Commitdccad6c

authored

Update single_node_scratch.sh

1 parent1d03a2a commitdccad6cCopy full SHA for dccad6c

File tree

-2

lines changed

-2

lines changed

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ vit_checkpoint_path="xxx/mae_pretrain_vit_base.pth" # downloaded from https://dr`
`17`	`17`	`### EXAMPLE ###`
`18`	`18`
`19`	`19`	`node=1`
`20`		`-node_num=1`
	`20`	`+node_num=8`
`21`	`21`	`torchrun --nnodes=${node} --nproc_per_node=${node_num} --master_port=10211 train.py \`
`22`	`22`	`--traj_cons \`
`23`	`23`	`--rgb_pad 10 \`
`@@ -57,4 +57,4 @@ torchrun --nnodes=${node} --nproc_per_node=${node_num} --master_port=10211 train`
`57`	`57`	`--warmup_epochs 5 \`
`58`	`58`	`--real_dataset_names${real_dataset_names} \`
`59`	`59`	`--use_aug_data \`
`60`		`- --report_to_wandb \`
	`60`	`+ --report_to_wandb \`

Comments

(0)