Update deep_inspect.py

ain-soph · May 26, 2022 · 22059ba · 22059ba
1 parent 2924c32
commit 22059ba
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):