Spaces:
Running
Running
File size: 619 Bytes
c4c7cee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
# @package _global_
defaults:
- override /dataset: osv5m_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-4
weight_decay: 0.05
dataset:
full_batch_size: 1024
trainer:
gradient_clip_val: 0.05
gradient_clip_algorithm: norm
experiment_name_suffix: von_fisher_mixture
|