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/sky/sky_sample.yaml \ --ckpt ./share_ckpts/skytimelapse.pt \ --save_video_path ./test
