d283e7d051a51f0108ff3e7d9b2ad4ca00f78184,autokeras/preprocessor.py,ImageDataTransformer,_transform,#ImageDataTransformer#Any#Any#Any#,171

Before Change

    def _transform(self, compose_list, data, targets):
        data = data / self.max_val
        data = torch.Tensor(data.transpose(0, 3, 1, 2))
        data_transforms = Compose(compose_list)
        return MultiTransformDataset(data, targets, data_transforms)

After Change

    def _transform(self, compose_list, data, targets):
        data = data / self.max_val
        args = [0, len(data.shape) - 1] + list(range(1, len(data.shape) - 1))
        data = torch.Tensor(data.transpose(*args))
        data_transforms = Compose(compose_list)
        return MultiTransformDataset(data, targets, data_transforms)

Instances