You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
13 lines
359 B
13 lines
359 B
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),)
|