import torch
import torchvision
import syft as sy

hook = sy.TorchHook(torch)
alice = sy.VirtualWorker(hook, id="alice")

dataset = torchvision.datasets.ImageFolder(
    root="./data/chest_xray/train",
    transform=torchvision.transforms.ToTensor()
)

data_loader = torch.utils.data.DataLoader(
    dataset,
    batch_size=5,
    shuffle=True
).federate(("alice"))