Fixed dendrite.py
@@ -207,10 +207,10 @@ class _RemoteModuleCall(torch.autograd.Function):
# Check batch_size.
# TODO(const) check sequence_len when images have sequence len.
- if output.size(0) != inputs.size(0):
+ if outputs.size(0) != inputs.size(0):
return torch.zeros((inputs.size(0), bittensor.__network_dim__))
- return output
+ return outputs
@staticmethod
@once_differentiable
