change(example/cifar): switch wd back

2020-07-11 08:51:09 +10:00 · 2020-07-11 08:51:09 +10:00 · 64c1f378c5
commit 64c1f378c5
parent 274f419f70
1 changed files with 1 additions and 2 deletions
--- a/example/cifar/main.go
+++ b/example/cifar/main.go
@ -120,10 +120,9 @@ func main() {

 	var bestAccuracy float64

-	batchSize := 512
 	for epoch := 0; epoch < 350; epoch++ {
 		// opt.SetLR(learningRate(epoch))
-		optConfig := nn.NewSGDConfig(0.9, 0.0, 5e-4*float64(batchSize), true)
+		optConfig := nn.NewSGDConfig(0.9, 0.0, 5e-4, true)
 		var opt nn.Optimizer
 		var err error
 		switch {