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

    

    trunk_output = current
    self.model_trunk = tf.keras.Model(inputs=sequence, outputs=trunk_output)

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

Instances