Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
3e0f0621
Commit
3e0f0621
authored
May 15, 2020
by
Julien Chaumond
Browse files
Fix addcmul_
parent
fc2a4c88
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
src/transformers/optimization.py
src/transformers/optimization.py
+1
-1
No files found.
src/transformers/optimization.py
View file @
3e0f0621
...
@@ -153,7 +153,7 @@ class AdamW(Optimizer):
...
@@ -153,7 +153,7 @@ class AdamW(Optimizer):
# Decay the first and second moment running average coefficient
# Decay the first and second moment running average coefficient
# In-place operations to update the averages at the same time
# In-place operations to update the averages at the same time
exp_avg
.
mul_
(
beta1
).
add_
(
grad
,
alpha
=
1.0
-
beta1
)
exp_avg
.
mul_
(
beta1
).
add_
(
grad
,
alpha
=
1.0
-
beta1
)
exp_avg_sq
.
mul_
(
beta2
).
addcmul_
(
grad
,
grad
,
1.0
-
beta2
)
exp_avg_sq
.
mul_
(
beta2
).
addcmul_
(
grad
,
grad
,
value
=
1.0
-
beta2
)
denom
=
exp_avg_sq
.
sqrt
().
add_
(
group
[
"eps"
])
denom
=
exp_avg_sq
.
sqrt
().
add_
(
group
[
"eps"
])
step_size
=
group
[
"lr"
]
step_size
=
group
[
"lr"
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment