physics
calorimeter
fast-simulation
generative-model
flow-matching
lorenzov506 commited on
Commit
6f789fa
·
verified ·
1 Parent(s): 8657080

Add config.yaml duplicate of conf.yaml (enable HF download counter)

Browse files
Files changed (1) hide show
  1. simplebox/config.yaml +37 -0
simplebox/config.yaml ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: pretrain_SimpleBox
2
+ result_path: .
3
+
4
+ data:
5
+ data_file: SimpleBox_pretraining_4M.h5
6
+ batch_size: 1024
7
+ batch_size_val: 4096
8
+ train_fraction: 0.975
9
+ max_samples: null
10
+ use_nlayers_conditioning: true
11
+ use_direction_conditioning: true
12
+ transform_num_points:
13
+ - [Log, {alpha: 0.5}]
14
+ - [StandardScaler, {shape: [1, 45]}]
15
+ transform_fsamp:
16
+ - [MinMaxScaler, {shape: [1, 1], target_min: -1.0, target_max: 1.0}]
17
+ transform_nlayers:
18
+ - [MinMaxScaler, {shape: [1, 1], target_min: -1.0, target_max: 1.0}]
19
+ # directions are unit vectors — used as-is (no transform needed)
20
+
21
+ model:
22
+ name: FullyConnected
23
+ dim_input: 45 # num layers
24
+ dim_condition: 6 # energy + sampling_fraction + n_layers + dir_x + dir_y + dir_z
25
+ dim_time: 6
26
+ hidden_dims: [128, 256, 512, 256, 128]
27
+
28
+ training:
29
+ epochs: 5000
30
+ test_every: 100
31
+
32
+ optimizer:
33
+ name: Adam
34
+ lr: 1.0e-4
35
+ scheduler:
36
+ name: OneCycleLR
37
+ pct_start: 0.5