Skip to content

Commit 363818c

Browse files
committed
Update config.py
1 parent eeb15b1 commit 363818c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

preprocess/config.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,18 +32,18 @@ def __init__(self):
3232
self.rnn_num_steps = int((self.win_len - self.rnn_step_len) / self.rnn_step_stride) + 1
3333
self.num_att_heads = 4
3434
# SAR train
35-
self.num_epochs = 10
35+
self.num_epochs = 15
3636
self.batch_size = 128
3737
self.lr = 1e-4
3838
self.neg_ratio = 0.2
3939
self.ckpt_step = 100
4040
self.summary_step = 20
4141
# picking config
42-
self.trig_thres = 0.5
42+
self.trig_thres = 0.3
4343
self.picker_batch_size = 20
4444
self.tp_dev = 1.5 # merge picks in different sliding win
4545
self.ts_dev = 1.5
4646
self.amp_win = [1,6] # sec pre-P & post-S for amp calc
4747
self.rm_glitch = True
4848
self.win_peak = 1
49-
self.amp_ratio_thres = [5,10,2] # Peak rm; P/P_tail; P/S
49+
self.amp_ratio_thres = [6,10,2] # Peak rm; P/P_tail; P/S

0 commit comments

Comments
 (0)