a b/catenets/models/diffpo/config/depmap.yaml
1
2
#type: args
3
# Standard setting: num_step50_lr0.00005_epochs500
4
dataset:
5
  data_name: "depmap"
6
7
train:
8
  epochs: 500 #500
9
  batch_size: 256
10
  lr: 0.001 #0.00005
11
  valid_epoch_interval: 50
12
13
diffusion:
14
  layers: 4  
15
  channels: 64 
16
  f_dim: 180
17
  cond_dim: 185 # conditional variable dimension
18
  hidden_dim: 128 
19
  side_dim: 33
20
  nheads: 2 #4
21
  diffusion_embedding_dim: 128
22
  beta_start: 0.0001
23
  beta_end: 0.5
24
  num_steps: 100
25
  schedule: "quad"
26
  mixed: False
27
28
model:
29
  is_unconditional: 0
30
  timeemb: 32
31
  featureemb: 32 #16
32
  target_strategy: "random"
33
  mixed: False