15ac1c2bb2627bb8fbb89b0b64709e9a50be02fd,torchsample/transforms/tensor_transforms.py,ToFile,__call__,#ToFile#,91

Before Change

    def __call__(self, x, y=None):
        np.save(os.path.join(self.root,"x_img-%i.npy"%self.counter), x.numpy())
        if y is not None:
            np.save(os.path.join(self.root,"y_img-%i.npy"%self.counter), y.numpy())
            self.counter += 1
            return x, y
        else:

After Change

        self.counter = 0

    def __call__(self, *inputs):
        for idx, _input in inputs:
            fpath = os.path.join(self.root, "img_%i_%i"%(self.counter, idx))
            np.save(fpath, _input.numpy())
        self.counter += 1
        return inputs

Instances