BoxImageLoader.py 文件源码

def execute(self):

        batch_size = 50
        train_loader = torch.utils.data.DataLoader(datasets.MNIST('data', train=True, download=True, transform=transforms.ToTensor()), batch_size=batch_size, shuffle=True)

        self.data = train_loader

        for port in self.outPorts:
            port.transferData(self.data)