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-09 13:20:30 +10:00
parent 69380e1456
commit f94e0e93ce
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
example/cifar/main.go
View File

@ -83,11 +83,11 @@ func fastResnet(p nn.Path) (retVal nn.SequentialT) {
func learningRate(epoch int) (retVal float64) {
switch {
case epoch < 50:
return 0.1
return 0.4
case epoch < 100:
return 0.01
return 0.2
default:
return 0.001
return 0.1
}
}
@ -125,10 +125,11 @@ func main() {
var lossVal float64
startTime := time.Now()
for epoch := 0; epoch < 150; epoch++ {
for epoch := 0; epoch < 24; epoch++ {
opt.SetLR(learningRate(epoch))
iter := ts.MustNewIter2(ds.TrainImages, ds.TrainLabels, int64(64))
// iter := ts.MustNewIter2(ds.TrainImages, ds.TrainLabels, int64(64))
iter := ts.MustNewIter2(ds.TrainImages, ds.TrainLabels, int64(512))
iter.Shuffle()
// iter = iter.ToDevice(device)