increase timeout for dgl.nn.NodeEmbedding TCPStore get/wait to 10 minutes (#2967)

Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com>

python/dgl/nn/pytorch/sparse_emb.py

@@ -82,7 +82,7 @@ class NodeEmbedding: # NodeEmbedding
             # embeding status synchronization across GPU processes
             if _STORE is None:
                 _STORE = th.distributed.TCPStore(
-                    host_name, port, world_size, True, timedelta(seconds=30))
+                    host_name, port, world_size, True, timedelta(seconds=10*60))
             for _ in range(1, world_size):
                 # send embs
                 _STORE.set(name, name)
@@ -90,7 +90,7 @@ class NodeEmbedding: # NodeEmbedding
             # receive
             if _STORE is None:
                 _STORE = th.distributed.TCPStore(
-                    host_name, port, world_size, False, timedelta(seconds=30))
+                    host_name, port, world_size, False, timedelta(seconds=10*60))
             _STORE.wait([name])
             emb = get_shared_mem_array(name, (num_embeddings, embedding_dim), th.float32)