weight_decay=args.adam_weight_decay,
    eps=args.adam_epsilon,
) Then it loads a dataset and you can specify how to preprocess it:   Copied dataset = load_dataset("imagefolder", data_dir=args.train_data_dir, cache_dir=args.cache_dir, split="train")

augmentations = transforms.Compose(
    [
        transforms.Resize(args.resolution, interpolation=transforms.InterpolationMode.BILINEAR),
        transforms.CenterCrop(args.resolution) if args.cen