画像を水増ししたデータを保存

Question

前提・実現したいこと

pytorch 画像データの水増し、保存

ここに質問の内容を詳しく書いてください。
３値画像分類で、画像データが少ないため、データの水増しをしましたが、
それらをそれぞれのフォルダに保存できないでおります。

該当のソースコード

dataset_root_dir = '/content/drive/MyDrive/train'
※trainフォルダの中に３フォルダそれぞれ五十枚画像データが入っております。

dataset = datasets.ImageFolder(root=dataset_root_dir)

data_transform = transforms.Compose([
    transforms.RandomHorizontalFlip(),
    transforms.RandomVerticalFlip(),
])

dataset_augmentated = datasets.ImageFolder(root=dataset_root_dir, transform=data_transform)

dataset_augmentated.save("/content/drive/MyDrive/train/".format(n+1))

試したこと

画像データの階層を変更
trainフォルダに１つのフォルダのみ設定

補足情報（FW/ツールのバージョンなど）

pytorch使用

解決策などありましたらご教授お願い致します。