From fe24caa1297efde78ed1453894d397e85aa4e123 Mon Sep 17 00:00:00 2001 From: wanghh2000 Date: Thu, 28 Dec 2023 10:59:38 +0800 Subject: [PATCH] Update use accelerate cmd --- ptuning/finetune-p-tuning-v2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ptuning/finetune-p-tuning-v2.py b/ptuning/finetune-p-tuning-v2.py index d294fd8..11539cd 100644 --- a/ptuning/finetune-p-tuning-v2.py +++ b/ptuning/finetune-p-tuning-v2.py @@ -19,9 +19,9 @@ Fine-tuning the library models for sequence to sequence for P-Tuning v2 # You can also adapt this script on your own sequence to sequence task. Pointers for this are left as comments. # CUDA_VISIBLE_DEVICES=-1 python finetune-p-tuning-v2.py - -# accelerate launch --cpu --num_machines=1 --num_processes=1 --num_cpu_threads_per_process=1 finetune-p-tuning-v2.py # accelerate launch --cpu --num_machines=1 --num_processes=4 --num_cpu_threads_per_process=1 finetune-p-tuning-v2.py +# accelerate launch --num_machines=1 --num_processes=1 --gpu_ids=1 finetune-p-tuning-v2.py + # import logging import os