Less Stressed Augmentation Tuning with Hydra — Why we need Hydra for augmentation? — How do you write augmentation code in your dataset loader? You must write it like this: from torch.utils.data import Dataset
from torchvision.transforms import transforms
class Dataset(Dataset):
def __init__(self, data_dir: str):
...
augmentations = transforms.Compose([
transforms.CenterCrop(224),
transforms.RandomAutocontrast(0.45),
transforms.Normalize(),
transforms.ToTensor() …
