fix bugs on start historical

examples/test_all.sh

@@ -8,9 +8,8 @@ partition_params=("ours" )
 #("ours" "metis" "ldg" "random")
 partitions="12"
 node_per="4"
-nnodes="3"
-node_rank="1"
-probability_params=("0.1")
+nnodes="2"
+node_rank="0"
 probability_params=("0.1")
 sample_type_params=("boundery_recent_decay")
 #sample_type_params=("recent" "boundery_recent_decay") #"boundery_recent_uniform")

examples/train_boundery.py

@@ -237,7 +237,7 @@ def main():
     if memory_param['type'] != 'none':
         mailbox = SharedMailBox(graph.ids.shape[0], memory_param, dim_edge_feat = graph.efeat.shape[1] if graph.efeat is not None else 0,
         shared_nodes_index=graph.shared_nids_list[ctx.memory_group_rank],device = torch.device('cuda:{}'.format(local_rank)),
-        start_historical=(args.memory_type=='historical')
+        start_historical=(args.memory_type=='historical'),
         shared_ssim=args.shared_memory_ssim)
     else:
         mailbox = None