# Default training/eval configuration

seed: 42
image_size: 224

resnet:
  backbone: resnet50
  embedding_dim: 512
  batch_size: 64
  epochs: 20
  lr: 0.001
  weight_decay: 0.0001
  use_imagenet_pretrained: true

vit:
  embedding_dim: 512
  num_layers: 6
  num_heads: 8
  ff_multiplier: 4
  dropout: 0.1
  batch_size: 64
  epochs: 30
  lr: 0.0005
  weight_decay: 0.05

paths:
  polyvore_root: /kaggle/input/polyvore-outfits
  resnet_ckpt: models/exports/resnet_item_embedder.pth
  vit_ckpt: models/exports/vit_outfit_model.pth
  export_dir: models/exports