Skip to content

Commit 0dff1eb

Browse files
authored
Merge pull request #13 from lucasb-eyer/patch-2
Fix typo in code
2 parents bc38912 + 9c977f6 commit 0dff1eb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sotabencheval/image_classification/imagenet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -276,8 +276,8 @@ def add(self, output_dict: dict):
276276

277277
if not self.first_batch_processed:
278278
hash_dict = output_dict
279-
hash_dict['Top 1 Accuracy'] = self.top1.avg,
280-
hash_dict['Top 5 Accuracy'] = self.top5.av
279+
hash_dict['Top 1 Accuracy'] = self.top1.avg
280+
hash_dict['Top 5 Accuracy'] = self.top5.avg
281281
self.batch_hash = calculate_batch_hash(hash_dict)
282282
self.first_batch_processed = True
283283

0 commit comments

Comments
 (0)