open_flamingo
open_flamingo copied to clipboard
Fix eval_imagenet and add eval_winoground
- Replace
.cuda()
calls with.to()
calls and correct device ineval_imagenet
function - Add
eval_winoground
to evaluate on the Winoground dataset