Merge pull request #133 from Hollen0318/Hollen

notes for dataset and threat model
RobustBench · Mar 28, 2023 · 2317b19 · 2317b19
2 parents 9585c1b + 3ff80ad
commit 2317b19
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/robustbench/leaderboard/template.py b/robustbench/leaderboard/template.py
@@ -38,8 +38,8 @@ def generate_leaderboard(dataset: Union[str, BenchmarkDataset],
     If the model is robust to common corruptions, then the "autoattack_acc" field should be
     "corruptions_acc".
 
-    :param dataset: The dataset of the wanted leaderboard.
-    :param threat_model: The threat model of the wanted leaderboard.
+    :param dataset: The dataset of the wanted leaderboard. example: "cifar10"
+    :param threat_model: The threat model of the wanted leaderboard. example: "Linf"
     :param models_folder: The base folder of the model jsons (e.g. our "model_info" folder).
 
     :return: The resulting HTML table.