71106ac70146d89b3e3e0774a5c27a98212e1405,basenji/seqnn.py,SeqNN,build_model,#SeqNN#Any#,82

Before Change

    //////////////////////////////////////////////////////////////////////////////////////////////////////

    slicegf = False
    if slicegf:
      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)


    if self.augment_rc:
      print("self.augment_rc")
      // transform back from reverse complement
      current = layers.SwitchReverse()([current, reverse_bool]) ////// needs to change for hic

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