Commit f9c80767 by xxx

modify static graph and merge1

parent 627d70b3
......@@ -122,7 +122,7 @@ void ParallelSampler :: neighbor_sample_from_nodes_static_layer(th::Tensor nodes
if(rst.second){ //不重复
eid_threads[tid].emplace_back(*chosen_e_iter);
node_s_threads[tid].insert(*chosen_n_iter);
if(temp_s.size()<fanout && !tnb.neighbors_set.empty() && temp_s.size()<tnb.neighbors_set[node].size()) fanout++;
if(!tnb.neighbors_set.empty() && temp_s.size()<fanout && temp_s.size()<tnb.neighbors_set[node].size()) fanout++;
}
}
else{
......
......@@ -99,7 +99,7 @@ def test():
is_distinct=0)
end = time.time()
print("init time:", end-pre)
# print(sampler.tnb.deg[0:10])
print(sampler.tnb.deg[0:100])
n_list = []
for i in range (g_data.num_nodes.item() // args.batch_size):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment