5c17417d4d48afb07629d2dfe014cb4b46a736c7,plugins/train/trainer/_base.py,Batcher,generate_preview,#Batcher#Any#,315

Before Change

        self.samples = batch[0]
        batch = batch[1:]   // Remove full size samples from batch
        if self.use_mask:
            batch = self.compile_mask(batch)
        self.target = batch[1]

    def set_preview_feed(self):
         Set the preview dictionary 

After Change

        batch = next(self.preview_feed)
        self.samples, feed = batch[:2]
        batch = batch[2:]   // Remove full size samples and feed from batch
        self.target = batch[self.model.largest_face_index]
        if self.use_mask:
            mask = batch[-1]
            batch = [[feed, mask], batch]
            self.target = [self.target, mask]

    def set_preview_feed(self):
         Set the preview dictionary 

Instances