OFA icon indicating copy to clipboard operation
OFA copied to clipboard

Update evaluate.py

Open JJJYmmm opened this issue 10 months ago • 0 comments

Fix the conditional determination of whether the scores variable is empty.

problem

When running evaluate_refcoco.sh, encounter errors below.

scores:tensor([1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1., 1.],
       device='cuda:0')
Traceback (most recent call last):
  File "/home/xxx/projects/OFA/run_scripts/refcoco/../../evaluate.py", line 188, in main
    if scores and isinstance(scores[0], tuple):
RuntimeError: Boolean value of Tensor with more than one value is ambiguous

Solution

            # if scores and isinstance(scores[0], tuple):
            if scores is not None and isinstance(scores[0], tuple):
                score_sum += sum([s[0] for s in scores])
                score_cnt += sum([s[1] for s in scores])

JJJYmmm avatar Apr 24 '24 06:04 JJJYmmm