Fix Wu2020Adversarial

dedeswim · dedeswim · commit f3f450ed68cb · 2021-05-04T10:08:48.000+02:00
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".'

Original file line number	Diff line number	Diff line change
`@@ -160,7 +160,7 @@ def _safe_load_state_dict(model: nn.Module, model_name: str,`
`160`	`160`	`"Wong2020Fast", "Hendrycks2020AugMix_WRN", "Hendrycks2020AugMix_ResNeXt",`
`161`	`161`	`"Kireev2021Effectiveness_Gauss50percent", "Kireev2021Effectiveness_AugMixNoJSD",`
`162`	`162`	`"Kireev2021Effectiveness_RLAT", "Kireev2021Effectiveness_RLATAugMixNoJSD",`
`163`		`- "Chen2020Efficient"`
	`163`	`+ "Chen2020Efficient", "Wu2020Adversarial"`
`164`	`164`	`}`
`165`	`165`
`166`	`166`	`failure_message = 'Missing key(s) in state_dict: "mu", "sigma".'`