Skip to content
This repository was archived by the owner on Jul 7, 2023. It is now read-only.

update multistep_optimizer for tensorflow gpu #1773

Merged
afrozenator merged 8 commits into from
Jun 16, 2020
Merged

update multistep_optimizer for tensorflow gpu #1773

Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
29aff51
update multistep
AgoloCuongHoang Dec 17, 2019
134b75d
add backend
AgoloCuongHoang Dec 20, 2019
55248ba
add dtypes
AgoloCuongHoang Dec 20, 2019
a1e0d38
correct indent
AgoloCuongHoang Dec 20, 2019
a4a1323
add cond in functions
AgoloCuongHoang Dec 27, 2019
015344a
fix a type of function name
AgoloCuongHoang Dec 27, 2019
1001966
rename files
AgoloCuongHoang Apr 16, 2020
9890379
upload original files
AgoloCuongHoang Apr 16, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
add backend
  • Loading branch information
@AgoloCuongHoang
AgoloCuongHoang committed Dec 20, 2019
commit 134b75d8ba72d9a934c7699f49226d8b87368e85
2 changes: 1 addition & 1 deletion tensor2tensor/utils/multistep_optimizer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@
from tensorflow.python.training import optimizer
from tensorflow.python.training import training_ops
from tensorflow.python.util.tf_export import tf_export

from tensorflow.keras import backend as K


class MultistepAdamOptimizer(optimizer.Optimizer):
Expand Down