Commit a6fb7bc7 by zhlj

add event count time and no pipeline

parent 9b36a8ec
...@@ -620,7 +620,7 @@ def main(): ...@@ -620,7 +620,7 @@ def main():
else: else:
mailbox.set_memory_all_reduce(index,memory,memory_ts,mail,mail_ts,reduce_Op = 'max', async_op = False,filter=None,set_remote=True,mode='all_reduce') mailbox.set_memory_all_reduce(index,memory,memory_ts,mail,mail_ts,reduce_Op = 'max', async_op = False,filter=None,set_remote=True,mode='all_reduce')
""" """
print(time_count.print())
torch.cuda.synchronize() torch.cuda.synchronize()
dist.barrier() dist.barrier()
time_prep = time.time() - epoch_start_time time_prep = time.time() - epoch_start_time
......
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