lxuechen/defun.py

Created August 3, 2018 19:39

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/lxuechen/bad7cd680b25d726b6a2600c636b5d52.js"></script>
Save lxuechen/bad7cd680b25d726b6a2600c636b5d52 to your computer and use it in GitHub Desktop.

Download ZIP

general defun

Raw

	tfe = tf.contrib.eager
	model.call = tfe.defun(model.call)
	model.compute_gradients = tfe.defun(model.compute_gradients)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment