remove learning_rule argument from Conv and Pool

cnrl · Mar 24, 2021 · 055429f · 055429f
1 parent 55cdb08
commit 055429f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cnsproject/network/connections.py b/cnsproject/network/connections.py
@@ -3,7 +3,7 @@
 """
 
 from abc import ABC, abstractmethod
-from typing import Optional, Union, Sequence
+from typing import Union, Sequence
 
 import torch
 
@@ -156,6 +156,7 @@ def __init__(
         super().__init__(
             pre=pre,
             post=post,
+            lr=lr,
             weight_decay=weight_decay,
             **kwargs
         )
@@ -212,6 +213,7 @@ def __init__(
         super().__init__(
             pre=pre,
             post=post,
+            lr=lr,
             weight_decay=weight_decay,
             **kwargs
         )
@@ -268,7 +270,6 @@ def __init__(
         super().__init__(
             pre=pre,
             post=post,
-            learning_rule=learning_rule,
             lr=lr,
             weight_decay=weight_decay,
             **kwargs
@@ -329,7 +330,6 @@ def __init__(
         super().__init__(
             pre=pre,
             post=post,
-            learning_rule=learning_rule,
             lr=lr,
             weight_decay=weight_decay,
             **kwargs