Fix Wu2020Adversarial

RobustBench · May 4, 2021 · f3f450e · f3f450e
1 parent 737c3a5
commit f3f450e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/robustbench/utils.py b/robustbench/utils.py
@@ -160,7 +160,7 @@ def _safe_load_state_dict(model: nn.Module, model_name: str,
         "Wong2020Fast", "Hendrycks2020AugMix_WRN", "Hendrycks2020AugMix_ResNeXt",
         "Kireev2021Effectiveness_Gauss50percent", "Kireev2021Effectiveness_AugMixNoJSD",
         "Kireev2021Effectiveness_RLAT", "Kireev2021Effectiveness_RLATAugMixNoJSD",
-        "Chen2020Efficient"
+        "Chen2020Efficient", "Wu2020Adversarial"
     }
 
     failure_message = 'Missing key(s) in state_dict: "mu", "sigma".'