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

chore(colab)

Browse Source
This commit is contained in:
sugarme 2020-07-05 14:14:40 +10:00
parent b70073f9a5
commit 19b4122347
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

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