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

chore(example/cifar): update

Browse Source
This commit is contained in:
sugarme 2020-07-08 20:34:29 +10:00
parent bb4e286ee5
commit 2d2bf65ddf
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
example/cifar/main.go
View File

@ -128,6 +128,8 @@ func main() {
opt.SetLR(learningRate(epoch))
iter := ts.MustNewIter2(ds.TrainImages, ds.TrainLabels, int64(64))
iter = iter.ToDevice(device)
// iter.Shuffle()
// iter = iter.ToDevice(device)