Merge branch 'main' of https://github.com/rahulxie/trojanzoo

ain-soph · Jun 5, 2022 · 79d7d2c · 79d7d2c
2 parents be34d52 + a73cc6f
commit 79d7d2c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trojanvision/defenses/backdoor/model_inspection/deep_inspect.py b/trojanvision/defenses/backdoor/model_inspection/deep_inspect.py
@@ -48,8 +48,8 @@ def __init__(self, defense_remask_epoch: int = 20, defense_remask_lr=0.01,
     def optimize_mark(self, label: int, **kwargs) -> tuple[torch.Tensor, float]:
         r"""
         Args:
-            **kwargs:
             label (int): The class label to optimize.
+            **kwargs: Any keyword argument (unused).
 
         Returns:
             (torch.Tensor, torch.Tensor):