diff --git a/bindsnet/encoding/encoders.py b/bindsnet/encoding/encoders.py index 111e939fd..6d11e4e48 100644 --- a/bindsnet/encoding/encoders.py +++ b/bindsnet/encoding/encoders.py @@ -115,3 +115,18 @@ def __init__(self, time: int, dt: float = 1.0, **kwargs): super().__init__(time, dt=dt, **kwargs) self.enc = encodings.rank_order + + +class DelayEncoder(Encoder): + def __init__(self, time: int, dt: float = 1.0, **kwargs): + # language=rst + """ + Creates a callable DelayEncoder which encodes as defined in + :code:`bindsnet.encoding.delay` + + :param time: Length of delay spike train per input variable. + :param dt: Simulation time step. + """ + super().__init__(time, dt=dt, **kwargs) + + self.enc = encodings.delay \ No newline at end of file diff --git a/examples/mnist/reservoir.py b/examples/mnist/reservoir.py index 4c5ef80b8..4096b94a9 100644 --- a/examples/mnist/reservoir.py +++ b/examples/mnist/reservoir.py @@ -185,7 +185,7 @@ def forward(self, x): optimizer = torch.optim.SGD(model.parameters(), lr=1e-4, momentum=0.9) # Training the Model -print("\n Training the read out") +print("\n Training the read out ") pbar = tqdm(enumerate(range(n_epochs))) for epoch, _ in pbar: avg_loss = 0