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-08 23:06:23 +10:00
parent 7c49412cb3
commit abce9b78dc
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/cifar/main.go
View File

@ -140,7 +140,7 @@ func main() {
devicedData := item.Data.MustTo(vs.Device(), true)
devicedLabel := item.Label.MustTo(vs.Device(), true)
bimages := vision.Augmentation(devicedData, true, 4, 8)
bimages := vision.Augmentation(devicedData, true, 0, 0)
logits := net.ForwardT(bimages, true)
// logits := net.ForwardT(devicedData, true)