search

lancopku / agent-backdoor-attacks

Code&Data for the paper "Watch Out for Your Agents! Investigating Backdoor Threats to LLM-Based Agents"
25 stars 1 forks source link

GPU requirement? #3

Closed Zhang-Henry closed 2 months ago

Zhang-Henry commented 2 months ago

Great work! How much GPU memory will finetune require? Are there any GPU requirements? Thanks for the answer.

keven980716 commented 2 months ago

All our experiments are conducted on 8 A40 (40G). But I guess 4 40G would also be sufficient to support the fine-tuning.