fix time count

examples/test_all.sh

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

examples/train_boundery.py

@@ -589,7 +589,7 @@ def main():
             mailbox.update_p2p_mail()
             start = time_count.start_gpu()
             ada_param.update_gnn_aggregate_time(ada_param.last_start_event_gnn_aggregate)
-            ada_param.update_parameter()
+            #ada_param.update_parameter()
             #torch.cuda.empty_cache()
 
             """