Merge pull request #1742 from chao1995/fix/del-optimizer

PyTorchClassifier: delete optimizer before calling set_param()
Trusted-AI · Jun 21, 2022 · e442608 · e442608
2 parents cb3605c + f1e44a2
commit e442608
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/art/estimators/classification/pytorch.py b/art/estimators/classification/pytorch.py
@@ -516,6 +516,7 @@ def clone_for_refitting(self) -> "PyTorchClassifier":  # lgtm [py/inheritance/in
         clone.reset()
         params = self.get_params()
         del params["model"]
+        del params["optimizer"]
         clone.set_params(**params)
         return clone