def forward(self, s0, s1):
    tensors = [self.preproc0(s0), self.preproc1(s1)]
    for node in self.nodes:
        cur_tensor = node(tensors)
        tensors.append(cur_tensor)
    output = torch.cat(tensors[2:], dim=1)
    return output
def forward(self, s0, s1):
    tensors = [self.preproc0(s0), self.preproc1(s1)]
    for node in self.nodes:
        cur_tensor = node(tensors)
        tensors.append(cur_tensor)
    output = torch.cat(tensors[2:], dim=1)
    return output