train_loader = torch.utils.data.DataLoader(
FFDIDataset(train_label['path'].head(1000), train_label['target'].head(1000),
transforms.Resize((256, 256)),
transforms.RandomHorizontalFlip(),
transforms.RandomVerticalFlip(),
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
), batch_size=40, shuffle=True, num_workers=4, pin_memory=True
val_loader = torch.utils.data.DataLoader(
FFDIDataset(val_label['path'].head(1000), val_label['target'].head(1000),
transforms.Resize((256, 256)),
transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
), batch_size=40, shuffle=False, num_workers=4, pin_memory=True