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

change(example/cifar): augmentation with crop 32, cutout 8

Browse Source
This commit is contained in:
sugarme 2020-07-11 08:42:41 +10:00
parent 290421a30f
commit 7f078159fd
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
example/cifar/main.go
View File

@ -156,7 +156,8 @@ 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, 4, 8)
bimages := vision.Augmentation(devicedData, true, 32, 8)
logits := net.ForwardT(bimages, true)