---
seed: 42

model:
  architecture: inceptionv3
  num_classes: 120
  pretrained: true

dataset:
  type: stanford_dogs_cropped
  validation_set_size: 0
  image_size: [299, 299]

transformations:
  RandomResizedCrop:
    size: [299, 299]
    scale: [0.85, 1]
    ratio: [1, 1]
  ColorJitter:
    brightness: 0.2
    contrast: 0.2
    saturation: 0.1
    hue: 0.1
  RandomHorizontalFlip:
    p: 0.5

optimizer:
  Adam:
    lr: 0.001
    betas: [0.9, 0.999]
    weight_decay: 0

lr_scheduler:
  MultiStepLR:
    milestones: [80]
    gamma: 0.1

training:
  num_epochs: 100
  batch_size: 128
  dataloader_num_workers: 8
  save_path: results/

rtpt:
  experiment_name: InceptionV3_Stanford_Dogs
  name_initials: XX

wandb:
  enable_logging: true
  args:
    project: model_inversion_evaluation
    name: Inception_V3_Stanford_Dogs
    save_code: true