Skip to content

Commit

Permalink
lower benchmark learning rate
Browse files Browse the repository at this point in the history
  • Loading branch information
Jun Ru Anderson committed Aug 19, 2020
1 parent 4c98bbf commit 0eb5ad3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion benchmarks/transformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,7 +134,7 @@ def make_model(device, ntokens):
p = Pipe(model, balance)

criterion = nn.CrossEntropyLoss()
lr = 0.01 # learning rate
lr = 0.0005 # learning rate
optimizer = Adam(p.parameters(), lr=lr, precision=Precision.PURE_FP16)

return p, criterion, optimizer
Expand Down

0 comments on commit 0eb5ad3

Please sign in to comment.