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

updated

Browse Source
This commit is contained in:
sugarme 2020-07-05 14:41:20 +10:00
parent 6bdf7d585c
commit 19d451be0a
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

@ -132,7 +132,7 @@ func main() {
fmt.Printf("max layer: %v\n", maxLayer)
styleLayers := net.ForwardAllT(styleImg, false, maxLayer)
// styleLayers := net.ForwardAllT(styleImg, false, maxLayer)
contentLayers := net.ForwardAllT(contentImg, false, maxLayer)
vs := nn.NewVarStore(device)