leabogensperger commited on
Commit
51dad6c
·
verified ·
1 Parent(s): 8495076

Upload vae/config.yaml with huggingface_hub

Browse files
Files changed (1) hide show
  1. vae/config.yaml +33 -0
vae/config.yaml ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ data: metadata/cell_lines_drug_compounds.csv
2
+ data_root: /path/to/your/dataset
3
+ experiment_name: microscopy_vae
4
+ model:
5
+ base_channels: 64
6
+ cell_lines:
7
+ - ONS76
8
+ - UW228
9
+ in_channels: 3
10
+ latent_channels: 4
11
+ name: microscopy_vae
12
+ preprocessing:
13
+ img_size: 256
14
+ project_name: microscopy_vae
15
+ training:
16
+ alpha_ssim: 0.6
17
+ batch_size: 32
18
+ beta: 1.0e-07
19
+ ema_decay: 0.999
20
+ epochs: 200
21
+ filtered_only: false
22
+ lam_rot: 0.0
23
+ learning_rate: 1e-4
24
+ log_interval: 5
25
+ target_concentrations:
26
+ - 0.0
27
+ - 0.065
28
+ - 0.125
29
+ - 0.25
30
+ - 0.5
31
+ - 1.0
32
+ warmup_ratio: 0.05
33
+ weight_decay: 1.0e-05