dataset toml

[general]
flip_aug = false
color_aug = false
keep_tokens_separator= "|||"
shuffle_caption = false
caption_tag_dropout_rate = 0
caption_extension = ".txt"

[[datasets]]
batch_size = 2
enable_bucket = true
resolution = [512, 512]

  [[datasets.subsets]]
  image_dir = "D:\\training\512female"
  conditioning_data_dir = "D:\\training\512femalemask"
  num_repeats = 1

  [[datasets.subsets]]
  image_dir = "D:\\training\512male"
  conditioning_data_dir = "D:\\training\512malemask"
  num_repeats = 2

  [[datasets.subsets]]
  image_dir = "D:\\training\512both"
  conditioning_data_dir = "D:\\training\512bothmask"
  num_repeats = 2

  [[datasets.subsets]]
  image_dir = "D:\\training\512reg"
  conditioning_data_dir = "D:\\training\512regmask"
  num_repeats = 1

[[datasets]]
batch_size = 2
enable_bucket = true
resolution = [768, 768]

  [[datasets.subsets]]
  image_dir = "D:\\training\768female"
  conditioning_data_dir = "D:\\training\768femalemask"
  num_repeats = 1

  [[datasets.subsets]]
  image_dir = "D:\\training\768male"
  conditioning_data_dir = "D:\\training\768malemask"
  num_repeats = 3

  [[datasets.subsets]]
  image_dir = "D:\\training\768both"
  conditioning_data_dir = "D:\\training\768bothmask"
  num_repeats = 3

  [[datasets.subsets]]
  image_dir = "D:\\training\768reg"
  conditioning_data_dir = "D:\\training\768regmask"
  num_repeats = 2

[[datasets]]
batch_size = 1
enable_bucket = true
resolution = [1024, 1024]

  [[datasets.subsets]]
  image_dir = "D:\\training\1024female"
  conditioning_data_dir = "D:\\training\1024femalemask"
  num_repeats = 1

  [[datasets.subsets]]
  image_dir = "D:\\training\1024male"
  conditioning_data_dir = "D:\\training\1024malemask"
  num_repeats = 2

  [[datasets.subsets]]
  image_dir = "D:\\training\1024both"
  conditioning_data_dir = "D:\\training\1024bothmask"
  num_repeats = 2

  [[datasets.subsets]]
  image_dir = "D:\\training\1024reg"
  conditioning_data_dir = "D:\\training\1024regmask"
  num_repeats = 1