Skip to content

Instantly share code, notes, and snippets.

@lxuechen

lxuechen/defun_optim.py

Created August 3, 2018 19:39
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save lxuechen/94629057497b19c9807ea9a6910eb781 to your computer and use it in GitHub Desktop.
Save lxuechen/94629057497b19c9807ea9a6910eb781 to your computer and use it in GitHub Desktop.
Download ZIP
defun optimizer
Raw
defun_optim.py
def apply_gradients(optimizer, gradients, variables, global_step=None):
optimizer.apply_gradients(
zip(gradients, variables), global_step=global_step)
apply_gradients = tfe.defun(apply_gradients)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment