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:46:09 +10:00
parent d392e55598
commit 7c43a59171
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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,7 +145,7 @@ func main() {
styleWeight := ts.FloatScalar(StyleWeight)
for stepIdx := 1; stepIdx <= int(TotalSteps); stepIdx++ {
inputVar := path.VarCopy("img", contentImg)
fmt.Printf("Input var: %v\n", inputVar)
inputLayers := net.ForwardAllT(inputVar, false, maxLayer)
// var sLoss ts.Tensor