andr3/gotch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore(example/neural-style-transfer): updated

Browse Source
This commit is contained in:
sugarme 2020-07-04 12:25:53 +10:00
parent ceeb95e2c5
commit 8fcda3e1f9
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
example/neural-style-transfer/main.go
View File

@ -169,12 +169,15 @@ func main() {
loss := lossMul.MustAdd(cLoss, true)
opt.BackwardStep(loss)
if (stepIdx % 1000) == 0 {
img := inputVar.MustShallowClone().MustDetach()
err := in.SaveImage(img, fmt.Sprintf("out%v.jpg", stepIdx))
if (stepIdx % 10) == 0 {
clone := inputVar.MustShallowClone()
img := clone.MustDetach()
clone.MustDrop()
err := in.SaveImage(img, fmt.Sprintf("../../data/neural-style-transfer/out%v.jpg", stepIdx))
if err != nil {
log.Fatal(err)
}
img.MustDrop()
}
fmt.Printf("Step %v ... Done. Loss %10.1f\n", stepIdx, loss.Values()[0])