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

chore(example/cifar): updated

Browse Source
This commit is contained in:
sugarme 2020-07-08 21:35:44 +10:00
parent 0355d6de37
commit e8429ee6c1
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
example/cifar/main.go
View File

@ -137,12 +137,15 @@ func main() {
break
}
bimages := vision.Augmentation(item.Data.MustTo(vs.Device(), true), true, 4, 8)
devicedData := item.Data.MustTo(vs.Device(), true)
devicedLabel := item.Label.MustTo(vs.Device(), true)
bimages := vision.Augmentation(devicedData, true, 4, 8)
logits := net.ForwardT(bimages, true)
// logits := net.ForwardT(item.Data.MustTo(vs.Device(), true), false)
loss := logits.CrossEntropyForLogits(item.Label.MustTo(vs.Device(), true))
loss := logits.CrossEntropyForLogits(devicedLabel)
opt.BackwardStep(loss)
lossVal = loss.Values()[0]
@ -150,6 +153,8 @@ func main() {
// logits.MustDrop()
// item.Data.MustDrop()
// item.Label.MustDrop()
devicedData.MustDrop()
devicedLabel.MustDrop()
bimages.MustDrop()
loss.MustDrop()