add time count

examples/test_all.sh

@@ -9,9 +9,9 @@ partition_params=("ours")
 partitions="8"
 node_per="4"
 nnodes="2"
-node_rank="0"
-probability_params=("0.1" "0.05" "0.01" "0")
-sample_type_params=("recent" "boundery_recent_decay")
+node_rank="1"
+probability_params=("0.05")
+sample_type_params=("boundery_recent_decay")
 #sample_type_params=("recent" "boundery_recent_decay") #"boundery_recent_uniform")
 #memory_type=("all_update" "p2p" "all_reduce" "historical" "local")
 memory_type=("all_update")

examples/train_boundery.py

@@ -295,7 +295,7 @@ def main():
                                         mode='train',
                                         queue_size = 200,
                                         mailbox = mailbox,
-                                        is_pipeline=True,
+                                        is_pipeline=False,
                                         use_local_feature = False,
                                         device = torch.device('cuda:{}'.format(local_rank)),
                                         probability=args.probability,