prevent wrapping of a network that already contains another wrapped module

Author: lucidrains

GAF_microbatch_pytorch/GAF.py

@@ -185,6 +185,7 @@ def __init__(
         super().__init__()
 
         self.net = net
+        assert not any([m for m in net.modules() if isinstance(m, GAFWrapper)]), 'GAF wrapper cannot contain another network that is already wrapped'
 
         self.exclude_from_filtering = set(exclude_from_filtering)
 

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "GAF-microbatch-pytorch"
-version = "0.0.3"
+version = "0.0.4"
 description = "Gradient Agreement Filtering"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }