chore(example/neural-style-transfer): twist to test Colab GPU

2020-07-05 10:02:32 +10:00 · 2020-07-05 10:02:32 +10:00 · 97b8f9a653
commit 97b8f9a653
parent 1eb4f57f6f
1 changed files with 1 additions and 1 deletions
--- a/example/neural-style-transfer/main.go
+++ b/example/neural-style-transfer/main.go
@ -172,7 +172,7 @@ func main() {
 		if (stepIdx % 10) == 0 {
 			clone := inputVar.MustShallowClone()
 			img := clone.MustDetach()
-			clone.MustDrop()
+			// clone.MustDrop()
 			err := in.SaveImage(img, fmt.Sprintf("../../data/neural-style-transfer/out%v.jpg", stepIdx))
 			if err != nil {
 				log.Fatal(err)