njucckevin / SeeClick

The model, data and code for the visual GUI Agent SeeClick
Apache License 2.0
139 stars 8 forks source link

The gpu cost when running finetune/finetune_lora_ds.sh #24

Closed zhengshuo1 closed 2 months ago

zhengshuo1 commented 2 months ago

Hello, may I ask about the gpu cost when running finetune_lora_ds.sh. I have a 49GB*4 machine, however, when I ran the shell, it appeared to be out of memory.

njucckevin commented 2 months ago

Hi,

Our fine-tuning takes about 24h on 8*A100(80G). I suppose 49G*4 GPU should be ok, referring to Qwen-VL's fine-tuning guide. You may need to reduce the micro-batch-size.

zhengshuo1 commented 2 months ago

Hi,

Our fine-tuning takes about 24h on 8A100(80G). I suppose 49G4 GPU should be ok, referring to Qwen-VL's fine-tuning guide. You may need to reduce the micro-batch-size.

thanks, when I reduce micro-batch-size to 1, it is ok. I have another question. If I use the parameter max-length, it is shown "Unknown parameter passed".

njucckevin commented 2 months ago

The parameters are defined in finetune/finetune_lora_ds.sh, which include --max-length. What is your fine-tuning command?

zhengshuo1 commented 2 months ago

The parameters are defined in finetune/finetune_lora_ds.sh, which include --max-length. What is your fine-tuning command?

sorry. I tried it again, and there was nothing wrong.