ddf8c858e17b69a6206750269232a1717cd1a348,examples/mxnet/dis_kvstore/client.py,,start_client,#Any#,35

Before Change

    if my_client.get_id() % args.num_worker == 0:
        my_client.set_partition_book(name="entity_embed", partition_book=partition)
    else:
        time.sleep(3)
        my_client.set_partition_book(name="entity_embed")

    my_client.print()

After Change

    my_client.barrier()

    for i in range(100):
        my_client.push(name="entity_embed", id_tensor=ID[my_client.get_machine_id()], data_tensor=mx.nd.array([[0.,0.,0.],[0.,0.,0.]]))

    my_client.barrier()

    if my_client.get_id() % args.num_worker == 0:
        res = my_client.pull(name="entity_embed", id_tensor=mx.nd.array([0,1,2,3,4,5,6,7], dtype="int64"))

Instances