Hugging Face's logo

Spaces:
nicolas-dufour
/
Plonk
Running

App Files Community
Plonk
File size: 570 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
 
# @package _global_

defaults:
  - override /dataset: osv5m_emb
  - override /model: von_fisher
  - override /model/network: geo_adaln_mlp_von_fisher
  - override /model/loss: von_fisher
  - override /model/val_sampler: von_fisher
  - override /model/test_sampler: von_fisher
  - _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