feat: ✨ Add geometric consistency local rule.

Also fix residual conv pass
janelia-cellmap · Apr 26, 2024 · 701cb24 · 701cb24
1 parent 13f6ce1
commit 701cb24
Show file tree

Hide file tree

Showing 5 changed files with 242 additions and 172 deletions.
diff --git a/src/leibnetz/nets/__init__.py b/src/leibnetz/nets/__init__.py
@@ -1,12 +1,13 @@
 from .attentive_scalenet import build_attentive_scale_net
 from .scalenet import build_scale_net
 from .unet import build_unet
-from .bio import (
+from .local_learning import (
     convert_to_bio,
     convert_to_backprop,
     HebbsRule,
     KrotovsRule,
     OjasRule,
+    GeometricConsistencyRule,
 )
 
 # from .resnet import build_resnet