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

chore(colab update

Browse Source
This commit is contained in:
sugarme 2020-07-05 13:41:56 +10:00
parent 124285c8f2
commit d392e55598
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -137,7 +137,7 @@ func main() {
vs := nn.NewVarStore(device)
path := vs.Root()
inputVar := path.VarCopy("img", contentImg)
// inputVar := path.VarCopy("img", contentImg)
opt, err := nn.DefaultAdamConfig().Build(vs, LearningRate)
if err != nil {
log.Fatal(err)
@ -145,6 +145,7 @@ func main() {
styleWeight := ts.FloatScalar(StyleWeight)
for stepIdx := 1; stepIdx <= int(TotalSteps); stepIdx++ {
inputVar := path.VarCopy("img", contentImg)
inputLayers := net.ForwardAllT(inputVar, false, maxLayer)
// var sLoss ts.Tensor
@ -172,7 +173,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)
@ -181,7 +182,7 @@ func main() {
}
fmt.Printf("Step %v ... Done. Loss %10.1f\n", stepIdx, loss.Values()[0])
// cLoss.MustDrop()
cLoss.MustDrop()
loss.MustDrop()
}