|
|
|
@ -49,7 +49,7 @@ func newOptimizer(paramWithConfigs ParameterWithConfig) *optimizer {
|
|
|
|
|
cbuffer = C.malloc(C.size_t(len(p.Content)))
|
|
|
|
|
C.memcpy(cbuffer, unsafe.Pointer(&p.Content[0]), C.size_t(len(p.Content)))
|
|
|
|
|
o.opt = C.paddle_create_optimizer((*C.uchar)(&c[0]), C.int(len(c)),
|
|
|
|
|
C.paddle_element_type(p.ElementType), cbuffer, C.int(len(p.Content)*C.sizeof_float),
|
|
|
|
|
C.paddle_element_type(p.ElementType), cbuffer, C.int(len(p.Content)/C.sizeof_float),
|
|
|
|
|
(*C.char)(nullPtr), 0)
|
|
|
|
|
return o
|
|
|
|
|
}
|
|
|
|
@ -57,7 +57,7 @@ func newOptimizer(paramWithConfigs ParameterWithConfig) *optimizer {
|
|
|
|
|
func (o *optimizer) GetWeights() []byte {
|
|
|
|
|
var buffer unsafe.Pointer
|
|
|
|
|
buffer_len := C.paddle_optimizer_get_weights(o.opt, &buffer)
|
|
|
|
|
return cArrayToSlice(buffer, int(buffer_len))
|
|
|
|
|
return cArrayToSlice(buffer, int(buffer_len)*C.sizeof_float)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
func (o *optimizer) UpdateParameter(g Gradient) error {
|
|
|
|
@ -65,7 +65,7 @@ func (o *optimizer) UpdateParameter(g Gradient) error {
|
|
|
|
|
return fmt.Errorf("Name: %s, parameter and gradient element type not match, parameter: %v, gradient: %v", g.Name, o.elementType, g.ElementType)
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
r := C.paddle_update_parameter(o.opt, C.paddle_element_type(g.ElementType), unsafe.Pointer(&g.Content[0]), C.int(len(g.Content)))
|
|
|
|
|
r := C.paddle_update_parameter(o.opt, C.paddle_element_type(g.ElementType), unsafe.Pointer(&g.Content[0]), C.int(len(g.Content))/C.sizeof_float)
|
|
|
|
|
if r != 0 {
|
|
|
|
|
return fmt.Errorf("optimizer update returned error code: %d", r)
|
|
|
|
|
}
|
|
|
|
|