search

mlfoundations / task_vectors

Editing Models with Task Arithmetic
436 stars 37 forks source link

number of gpus for finetune #17

Open jihwanp opened 3 months ago

jihwanp commented 3 months ago

hi It seems that in finetune.py, all the available GPUs on the node are used for fine-tuning the models. In that case, did you use 8 GPUs or a different number?

Thanks