segm.yaml 566 Bytes
settings:
  root_dir: "./datasets/celeba/images/"
  train_anns: "./datasets/celeba/annotations/train.csv"
  val_anns: "./datasets/celeba/annotations/val.csv"

  checkpoint_path: "weights" #"/content/drive/MyDrive/weights/Places365 Inpainting/unet/phase 1"
  sample_folder: "sample" #"/content/drive/MyDrive/results/Places365 Inpainting/unet/phase 1"

  cuda: True
  lr: 0.001
  batch_size: 4
  num_workers: 4

  step_iters: [50000, 75000, 100000]
  gamma: 0.1

  visualize_per_iter: 1000
  save_per_iter: 1000
  print_per_iter: 10
  num_epochs: 100

  img_size: 512