13 lines
359 B
Python
13 lines
359 B
Python
|
import numpy
|
||
|
|
|
||
|
|
from thinc.backends._param_server import ParamServer
|
||
|
|
|
||
|
|
|
||
|
|
def test_param_server_init():
|
||
|
|
array = numpy.zeros((5,), dtype="f")
|
||
|
|
params = {("a", 1): array, ("b", 2): array}
|
||
|
|
grads = {("a", 1): array, ("c", 3): array}
|
||
|
|
ps = ParamServer(params, grads)
|
||
|
|
assert ps.param_keys == (("a", 1), ("b", 2))
|
||
|
|
assert ps.grad_keys == (("a", 1),)
|