File tree Expand file tree Collapse file tree 4 files changed +6
-10
lines changed Expand file tree Collapse file tree 4 files changed +6
-10
lines changed Original file line number Diff line number Diff line change @@ -4,14 +4,12 @@ data_url = 'ftp.mcs.anl.gov/pub/candle/public/benchmarks/Pilot1/uno/'
4
4
savepath = './results'
5
5
log_interval = 10
6
6
train_data = 'top_21_auc_1fold.uno.h5'
7
- learning_rate = 0.01
7
+ learning_rate = 0.025
8
8
learning_rate_min = 0.001
9
9
momentum = 0.9
10
10
weight_decay = 3e-4
11
11
grad_clip = 5
12
12
batch_size = 100
13
13
epochs = 10
14
14
seed = 13
15
- lr = 0.025
16
- lr_min = 0.001
17
15
Original file line number Diff line number Diff line change @@ -77,15 +77,15 @@ def run(params):
77
77
78
78
optimizer = optim .SGD (
79
79
model .parameters (),
80
- args .lr ,
80
+ args .learning_rate ,
81
81
momentum = args .momentum ,
82
82
weight_decay = args .weight_decay
83
83
)
84
84
85
85
scheduler = optim .lr_scheduler .CosineAnnealingLR (
86
86
optimizer ,
87
87
float (args .epochs ),
88
- eta_min = args .lr_min
88
+ eta_min = args .learning_rate_min
89
89
)
90
90
91
91
train_meter = darts .EpochMeter (tasks , 'train' )
Original file line number Diff line number Diff line change @@ -4,14 +4,12 @@ data_url = 'http://ftp.mcs.anl.gov/pub/candle/public/benchmarks/Pilot1/uno/'
4
4
savepath = '.'
5
5
log_interval = 10
6
6
train_data = 'top_21_auc_1fold.uno.h5'
7
- learning_rate = 0.01
7
+ learning_rate = 0.025
8
8
learning_rate_min = 0.001
9
9
momentum = 0.9
10
10
weight_decay = 3e-4
11
11
grad_clip = 5
12
12
batch_size = 100
13
13
epochs = 10
14
14
seed = 13
15
- lr = 0.025
16
- lr_min = 0.001
17
15
Original file line number Diff line number Diff line change @@ -56,15 +56,15 @@ def run(params):
56
56
57
57
optimizer = optim .SGD (
58
58
model .parameters (),
59
- args .lr ,
59
+ args .learning_rate ,
60
60
momentum = args .momentum ,
61
61
weight_decay = args .weight_decay
62
62
)
63
63
64
64
scheduler = optim .lr_scheduler .CosineAnnealingLR (
65
65
optimizer ,
66
66
float (args .epochs ),
67
- eta_min = args .lr_min
67
+ eta_min = args .learning_rate_min
68
68
)
69
69
70
70
train_meter = darts .EpochMeter (tasks , 'train' )
You can’t perform that action at this time.
0 commit comments