We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1778ddb commit 623fd3dCopy full SHA for 623fd3d
bindsnet/pipeline/environment_pipeline.py
@@ -252,10 +252,7 @@ def step_(
252
obs = obs.unsqueeze(0).unsqueeze(0)
253
obs_shape = torch.tensor([1] * len(obs.shape[1:]), device=self.device)
254
inputs = {
255
- k: self.encoding(
256
- obs.repeat(self.time, *obs_shape).to(self.device),
257
- device=self.device,
258
- )
+ k: obs.repeat(self.time, *obs_shape).to(self.device)
259
for k in self.inputs
260
}
261
else:
0 commit comments