Running out of GPU using summarize_emails.py with batch size 8

IST-DASLab / PanzaMail

Apache License 2.0

255 stars 13 forks source link

Running out of GPU using summarize_emails.py with batch size 8 #6

Closed Preetika764 closed 4 months ago

Preetika764 commented 4 months ago

gpu_error

Synthetic dataset works great but I'm running out of GPU while using a slightly bigger dataset of 293 personal emails. Reducing batch size to 4 in scripts/config.sh file helped to fix this issue.