emrectn/compute_grads.py

Created April 5, 2020 12:53

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

Select an option

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

Download ZIP

Revisions

emrectn created this gist Apr 5, 2020.

compute_grads.py

-Original file line number
+Diff line change
@@ -0,0 +1,6 @@
+    def compute_grads(cfg):
+      with tf.GradientTape() as tape:
+        all_loss = compute_loss(**cfg)
+      # Compute gradients wrt input image
+      total_loss = all_loss[0]
+      return tape.gradient(total_loss, cfg['init_image']), all_loss