Closed a1123622944 closed 1 week ago
执行该命令时候,内存使用过多 !CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 torchrun --nnodes=1 --nproc_per_node=8 \ inference_stage1.py --config configs/stage1-hand.yaml --output data/hand_example/hand_chip/repair \ --ckpt checkpoint/stage1_hand/checkpoint-stage1-hand.ckpt
colab环境 A100
执行该命令时候,内存使用过多 !CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 torchrun --nnodes=1 --nproc_per_node=8 \ inference_stage1.py --config configs/stage1-hand.yaml --output data/hand_example/hand_chip/repair \ --ckpt checkpoint/stage1_hand/checkpoint-stage1-hand.ckpt