test.yml

base: config/base_episode.yml

model:
  backbone: resnet50
  pooling: gem_pooling
  head: BNHead
  bn_where: after

data:
  name: pa_100k
  image_size: [256, 128]
  train:
    num_attribute: 8
    num_positive: 8
    num_negative: 8
    num_iterator: 2000
  val:
    num_attribute: 8
    num_positive: 8
    num_negative: 8
    num_iterator: 100

optimizer:
  name: sgd
  lr: 0.01
  specified_lr:
    enable: True
    lr: 0.1
    layers: [head]

loss:
  name: Singular_BCE
  reduction: mean

freeze:
  enable: True
  layers: [backbone]
  epochs: 10

lr_scheduler:
  enable: True
  name: ReduceLROnPlateau
  factor: 0.1
  patience: 5
  min_lr: 0

iters_to_accumulate: 2

clip_grad_norm_:
  enable: True
  max_norm: 10.0

trainer:
  epochs: 120
    
trainer_colab:
  epochs: 120