"remove unused debug info"

gangliao-patch-1
dongzhihong 8 years ago
parent 42dcffc29c
commit 6bb84b963d

@ -2,6 +2,7 @@ package pserver
import ( import (
"io/ioutil" "io/ioutil"
"reflect"
"testing" "testing"
) )
@ -22,3 +23,26 @@ func TestOptimizerCreateRelease(t *testing.T) {
o := newOptimizer(param) o := newOptimizer(param)
o.Cleanup() o.Cleanup()
} }
func TestOptimizerFull(t *testing.T) {
p := Parameter{
Name: "a",
ElementType: Float32,
}
p.Content = []byte{1, 3}
config, err := ioutil.ReadFile("./cclient/test/testdata/optimizer.pb.txt")
if err != nil {
t.Fatalf("read optimizer proto failed")
}
param := ParameterWithConfig{
Param: p,
Config: config,
}
o := newOptimizer(param)
g := Gradient(p)
if !reflect.DeepEqual(p.Content, o.GetWeights()) {
t.FailNow()
}
o.UpdateParameter(g)
o.Cleanup()
}

Loading…
Cancel
Save