Skip to content

Commit 7f1e99b

Browse files
committed
Use learning_rate and learning_rate_min
This keeps things consisten with CANDLE conventions, and removes potentially confusing parameters.
1 parent b986926 commit 7f1e99b

File tree

4 files changed

+6
-10
lines changed

4 files changed

+6
-10
lines changed

examples/darts/advanced/default_model.txt

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,14 +4,12 @@ data_url = 'ftp.mcs.anl.gov/pub/candle/public/benchmarks/Pilot1/uno/'
44
savepath = './results'
55
log_interval = 10
66
train_data = 'top_21_auc_1fold.uno.h5'
7-
learning_rate = 0.01
7+
learning_rate = 0.025
88
learning_rate_min = 0.001
99
momentum = 0.9
1010
weight_decay = 3e-4
1111
grad_clip = 5
1212
batch_size = 100
1313
epochs = 10
1414
seed = 13
15-
lr = 0.025
16-
lr_min = 0.001
1715

examples/darts/advanced/example.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -77,15 +77,15 @@ def run(params):
7777

7878
optimizer = optim.SGD(
7979
model.parameters(),
80-
args.lr,
80+
args.learning_rate,
8181
momentum=args.momentum,
8282
weight_decay=args.weight_decay
8383
)
8484

8585
scheduler = optim.lr_scheduler.CosineAnnealingLR(
8686
optimizer,
8787
float(args.epochs),
88-
eta_min=args.lr_min
88+
eta_min=args.learning_rate_min
8989
)
9090

9191
train_meter = darts.EpochMeter(tasks, 'train')

examples/darts/uno/default_model.txt

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,14 +4,12 @@ data_url = 'http://ftp.mcs.anl.gov/pub/candle/public/benchmarks/Pilot1/uno/'
44
savepath = '.'
55
log_interval = 10
66
train_data = 'top_21_auc_1fold.uno.h5'
7-
learning_rate = 0.01
7+
learning_rate = 0.025
88
learning_rate_min = 0.001
99
momentum = 0.9
1010
weight_decay = 3e-4
1111
grad_clip = 5
1212
batch_size = 100
1313
epochs = 10
1414
seed = 13
15-
lr = 0.025
16-
lr_min = 0.001
1715

examples/darts/uno/uno_example.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,15 +56,15 @@ def run(params):
5656

5757
optimizer = optim.SGD(
5858
model.parameters(),
59-
args.lr,
59+
args.learning_rate,
6060
momentum=args.momentum,
6161
weight_decay=args.weight_decay
6262
)
6363

6464
scheduler = optim.lr_scheduler.CosineAnnealingLR(
6565
optimizer,
6666
float(args.epochs),
67-
eta_min=args.lr_min
67+
eta_min=args.learning_rate_min
6868
)
6969

7070
train_meter = darts.EpochMeter(tasks, 'train')

0 commit comments

Comments
 (0)