We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4af85dc commit 7f3e7c8Copy full SHA for 7f3e7c8
1 file changed
ngclearn/components/neurons/graded/gaussianErrorCell.py
@@ -41,7 +41,7 @@ def __init__(self, name, n_units, batch_size=1, sigma=1., shape=None, **kwargs):
41
else:
42
_shape = (batch_size, shape[0], shape[1], shape[2]) ## shape is 4D tensor
43
sigma_shape = (1,1)
44
- if not isinstance(sigma, float):
+ if not isinstance(sigma, float) and not isinstance(sigma, int):
45
sigma_shape = jnp.array(sigma).shape
46
self.sigma_shape = sigma_shape
47
self.shape = shape
0 commit comments