Spaces:
Running
Running
File size: 217 Bytes
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 \
|