rwightman HF staff commited on
Commit
6316ef5
·
verified ·
1 Parent(s): af2dd62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,8 +58,8 @@ def load_leaderboard():
58
  result['infer_tflop_s'] = result['infer_samples_per_sec'] * result['infer_gmacs'] * 2 / 1000
59
 
60
  # Calculate average scores
61
- top1_columns = [col for col in result.columns if col.endswith('_top1')]
62
- top5_columns = [col for col in result.columns if col.endswith('_top5')]
63
  result['avg_top1'] = result[top1_columns].mean(axis=1)
64
  result['avg_top5'] = result[top5_columns].mean(axis=1)
65
 
 
58
  result['infer_tflop_s'] = result['infer_samples_per_sec'] * result['infer_gmacs'] * 2 / 1000
59
 
60
  # Calculate average scores
61
+ top1_columns = [col for col in result.columns if col.endswith('_top1') and not col == 'a_top1']
62
+ top5_columns = [col for col in result.columns if col.endswith('_top5') and not col == 'a_top5']
63
  result['avg_top1'] = result[top1_columns].mean(axis=1)
64
  result['avg_top5'] = result[top5_columns].mean(axis=1)
65