Skip to content

Commit ee4e401

Browse files
committed
minor cleanup of hebb syn
1 parent 55eea4c commit ee4e401

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

ngclearn/components/synapses/hebbian/hebbianSynapse.py

+9-9
Original file line numberDiff line numberDiff line change
@@ -220,6 +220,15 @@ def _reset(batch_size, shape):
220220
jnp.zeros(shape[1]), # db
221221
)
222222

223+
@resolver(_reset)
224+
def reset(self, inputs, outputs, pre, post, dWeights, dBiases):
225+
self.inputs.set(inputs)
226+
self.outputs.set(outputs)
227+
self.pre.set(pre)
228+
self.post.set(post)
229+
self.dWeights.set(dWeights)
230+
self.dBiases.set(dBiases)
231+
223232
@classmethod
224233
def help(cls): ## component help function
225234
properties = {
@@ -265,15 +274,6 @@ def help(cls): ## component help function
265274
"hyperparameters": hyperparams}
266275
return info
267276

268-
@resolver(_reset)
269-
def reset(self, inputs, outputs, pre, post, dWeights, dBiases):
270-
self.inputs.set(inputs)
271-
self.outputs.set(outputs)
272-
self.pre.set(pre)
273-
self.post.set(post)
274-
self.dWeights.set(dWeights)
275-
self.dBiases.set(dBiases)
276-
277277
def __repr__(self):
278278
comps = [varname for varname in dir(self) if Compartment.is_compartment(getattr(self, varname))]
279279
maxlen = max(len(c) for c in comps) + 5

0 commit comments

Comments
 (0)