Added check statement
@@ -138,6 +138,7 @@ class Dendrite(nn.Module):
results = []
logger.info("NEURONS:{}".format(neurons))
for idx, neuron in enumerate(neurons):
+ if idx < len(x):
forward_inputs = x[idx]
# Get or create remote_neuron.
