09b8381f7dd52c67fdd629825ff319271167d8c5,basenji/seqnn.py,SeqNN,build_model,#SeqNN#Any#,82

Before Change

    // heads
    //////////////////////////////////////////////////////////////////////////////////////////////////////

    if not isinstance(self.head, list):
        self.head = [self.head]

    self.head_output = []

    for hi, head in enumerate(self.head):
        if not isinstance(head, list):

After Change

    // heads
    //////////////////////////////////////////////////////////////////////////////////////////////////////

    head_keys = natsorted([v for v in vars(self) if v.startswith("head")])
    self.heads = [getattr(self, hk) for hk in head_keys]

    self.head_output = []
    for hi, head in enumerate(self.heads):
      if not isinstance(head, list):

Instances