HellaSwag: speed up by parallelizing log-prob evaluation (#5020)
For Mistral-7B and fp16, time on my system goes down from 536 seconds to 423 seconds for the full evaluation dataset (10042 tasks). Co-authored-by: Iwan Kawrakow <iwan.kawrakow@gmail.com>
K
Kawrakow committed
3e945cc1e9c06d2001031360e4e303e9548fb02c
Parent: ad19812
Committed by GitHub <noreply@github.com>
on 1/18/2024, 5:18:21 PM