单卡运行命令

[UntrainedButLoveCode]

export OMP_NUM_THREADS=4 torchrun --nproc_per_node=1 --master_port 5228 train_distributed.py --gpu_id '0'
--gray_aug --gray_p 0.3 --scale_aug --scale_type 1 --scale_p 0.3 --epochs 1000 --lr_step 1200 --lr 1e-5
--batch_size 16 --num_patch 1 --threshold 0.35 --test_per_epoch 20
--dataset low --crop_size 256 --pre None --test_patch --save 用这段代码可以指定但卡跑，--nproc_per_node=1配合--gpu_id '0'使用。 --nproc_per_node=2 和train_distributed.py --gpu_id '0，1'双卡跑。

[3207779341]

单卡视频demo怎么写

[UntrainedButLoveCode]

单卡视频demo怎么写

这个我就不知道了