search

Wangbenzhi / RealisHuman

Code of RealisHuman: A Two-Stage Approach for Refining Malformed Human Parts in Generated Images
Apache License 2.0
50 stars 4 forks source link

内存使用过多 #8

Closed a1123622944 closed 1 week ago

a1123622944 commented 3 weeks ago

image 执行该命令时候，内存使用过多 !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

a1123622944 commented 3 weeks ago

colab环境 A100