ho22joshua commited on
Commit
da69013
·
1 Parent(s): df03c6f

fixing configs

Browse files
root_gnn_dgl/configs/stats_100K/finetuning_ttH_CP_even_vs_odd.yaml CHANGED
@@ -23,7 +23,7 @@ Model:
23
  Training:
24
  epochs: 500
25
  batch_size: 1024
26
- learning_rate: 0.0001
27
  gamma: 0.99
28
  Datasets:
29
  ttH_CP_even: &dataset_defn
 
23
  Training:
24
  epochs: 500
25
  batch_size: 1024
26
+ learning_rate: 0.00001
27
  gamma: 0.99
28
  Datasets:
29
  ttH_CP_even: &dataset_defn
root_gnn_dgl/configs/stats_all/finetuning_ttH_CP_even_vs_odd.yaml CHANGED
@@ -20,6 +20,11 @@ Model:
20
  n_layers: 4
21
  n_proc_steps: 4
22
  dropout: 0
 
 
 
 
 
23
  Datasets:
24
  ttH_CP_even: &dataset_defn
25
  module: root_gnn_base.dataset
 
20
  n_layers: 4
21
  n_proc_steps: 4
22
  dropout: 0
23
+ Training:
24
+ epochs: 500
25
+ batch_size: 1024
26
+ learning_rate: 0.00001
27
+ gamma: 0.99
28
  Datasets:
29
  ttH_CP_even: &dataset_defn
30
  module: root_gnn_base.dataset