faabf5663a0d8fbdee527c03d396106dbd855270,alignment/data.py,FaceSegIter,reset,#FaceSegIter#,188

Before Change

        return D

    def reset(self):
      print("reset")
      Resets the iterator to the beginning of the data.
      if self.aug_level>0:
        random.shuffle(self.seq)
      self.cur = 0

After Change

    def reset(self):
      //print("reset")
      if self.aug_level==0:
        self.seq = self.oseq
      else:
        self.seq = []
        for _ in range(self.exf):
          _seq = self.oseq[:]
          random.shuffle(_seq)
          self.seq += _seq
        print("train size after reset", len(self.seq))
      self.cur = 0

    def next_sample(self):
      Helper function for reading in next sample.

Instances