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

Before Change

    current_length = current.shape[1]
    target_diff = self.target_length - current_length
    target_diff2 = target_diff // 2
    if target_diff2 < 0:
      print("Model over-pools to %d for target_length %d." % \
        (current_length, self.target_length), file=sys.stderr)
      exit(1)
    elif target_diff2 > 0:
      current = layers.SliceCenter(
        left=target_diff2,
        right=current_length-target_diff2)(current)

    self.trunk_output = current

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

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):

Instances