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:26:21 +10:00
parent f94e0e93ce
commit 9666d1c44e
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
example/cifar/main.go
View File

@ -117,7 +117,7 @@ func main() {
net := fastResnet(vs.Root())
optConfig := nn.NewSGDConfig(0.9, 0.0, 5e-4, true)
opt, err := optConfig.Build(vs, 0.0)
opt, err := optConfig.Build(vs, 0.2)
if err != nil {
log.Fatal(err)
}
@ -126,7 +126,7 @@ func main() {
startTime := time.Now()
for epoch := 0; epoch < 24; epoch++ {
opt.SetLR(learningRate(epoch))
// opt.SetLR(learningRate(epoch))
// iter := ts.MustNewIter2(ds.TrainImages, ds.TrainLabels, int64(64))
iter := ts.MustNewIter2(ds.TrainImages, ds.TrainLabels, int64(512))