configs/exp/YFCC100M_geoadalnmlp_von_fisher_mixture.yaml

# @package _global_

defaults:
  - override /dataset: yfcc_emb
  - override /model: von_fisher_mixture
  - override /model/network: geo_adaln_mlp_von_fisher_mixture
  - override /model/loss: von_fisher_mixture
  - override /model/val_sampler: von_fisher_mixture
  - override /model/test_sampler: von_fisher_mixture
  - _self_

model:
  network:
    depth: 11 # To compensate the increase in params
    dim: 512
  optimizer:
    optim:
      lr: 1e-5
      weight_decay: 0.05
dataset:
  full_batch_size: 1024
trainer:
  gradient_clip_val: 0.01
  gradient_clip_algorithm: norm
experiment_name_suffix: von_fisher_mixture
areas: []