94bafa8
1
2
3
4
5
6
7
8
#!/bin/bash export CUDA_VISIBLE_DEVICES=6,7 torchrun --nnodes=1 --nproc_per_node=2 sample/sample_ddp.py \ --config ./configs/taichi/taichi_sample.yaml \ --ckpt ./share_ckpts/taichi-hd.pt \ --save_video_path ./test \
