-
Notifications
You must be signed in to change notification settings - Fork 5
/
config.yml
47 lines (42 loc) · 914 Bytes
/
config.yml
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
# general
device: cuda
logging: False
seed: 3008
# model
model: sgspcn
upscale_factor: 12
rescale_factor: 1
input_type: rf
skip_bmode: True
das_b4_temporal: False
model_file: crimson-frost
# training
batch_size: 1
lr: 1e-4
epochs: 40
clutter_db: Null #-50
fine_tune: False
angle_threshold: Null
# loss
lambda0: 120
lambda1: 1
radial_sym_opt: False
# paths
data_path: PALA_data_InSilicoFlow/ #Rat18_2D_InVivoBrain #
ubx_dir: /storage/workspaces/artorg_aimi/ws_00000/chris/${data_path}
loc_dir: /home/chris/LocalDatasets/03_PALA/${data_path}
map_dir: /home/chris/UbelixWorkspaces/artorg_aimi/ws_00000/chris/${data_path}
data_dir: ${map_dir} #${ubx_dir} #
sequences: [0] #, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]
# inference config
channel_num: 143
nms_size: 11
nms_threshold: 20
gamma: 0.9
dither: False
eps: .5
pow_law_opt: False
wv_idcs: [0,1,2]
track: all_in #vel_z
synth_gt: False
save_image: False