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

Before Change

    my_client.barrier()

    my_client.push(name="entity_embed", id_tensor=ID[my_client.get_machine_id()], data_tensor=th.tensor([[0.,0.,0.],[0.,0.,0.]]))

    my_client.barrier()

    if my_client.get_id() % args.num_worker == 0:

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=th.tensor([[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=th.tensor([0,1,2,3,4,5,6,7]))

Instances