3090 が2枚ある環境で、マルチGPUを活用したい。
torchrun --standalone --nnodes=1 --nproc_per_node=2 train_ms_jp_extra.py -c Data/<model_name>/config.json -m Data/<model_name>
3090 が2枚ある環境で、マルチGPUを活用したい。
torchrun --standalone --nnodes=1 --nproc_per_node=2 train_ms_jp_extra.py -c Data/<model_name>/config.json -m Data/<model_name>
コメント