We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents bc38912 + 9c977f6 commit 0dff1ebCopy full SHA for 0dff1eb
sotabencheval/image_classification/imagenet.py
@@ -276,8 +276,8 @@ def add(self, output_dict: dict):
276
277
if not self.first_batch_processed:
278
hash_dict = output_dict
279
- hash_dict['Top 1 Accuracy'] = self.top1.avg,
280
- hash_dict['Top 5 Accuracy'] = self.top5.av
+ hash_dict['Top 1 Accuracy'] = self.top1.avg
+ hash_dict['Top 5 Accuracy'] = self.top5.avg
281
self.batch_hash = calculate_batch_hash(hash_dict)
282
self.first_batch_processed = True
283
0 commit comments