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

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

@ -145,7 +145,7 @@ func main() {
styleWeight := ts.FloatScalar(StyleWeight)
for stepIdx := 1; stepIdx <= int(TotalSteps); stepIdx++ {
fmt.Printf("Input var: %v\n", inputVar)
fmt.Printf("Input var size: %v\n", inputVar.MustSize())
inputLayers := net.ForwardAllT(inputVar, false, maxLayer)
// var sLoss ts.Tensor