From 8aa4bb40aa132787cf057bd3de82c7bcfa7afaee Mon Sep 17 00:00:00 2001 From: shu5620 Date: Sun, 1 Oct 2023 16:46:10 +0900 Subject: [PATCH] measure get_buffer_for_opt --- border-async-trainer/src/async_trainer/base.rs | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/border-async-trainer/src/async_trainer/base.rs b/border-async-trainer/src/async_trainer/base.rs index 6267512f..8384a1f1 100644 --- a/border-async-trainer/src/async_trainer/base.rs +++ b/border-async-trainer/src/async_trainer/base.rs @@ -307,9 +307,21 @@ where info!("Starts training loop"); loop { - let record = agent.opt(&mut async_buffer.get_buffer_for_opt().lock().unwrap()); + let record = { + let time_tmp = SystemTime::now(); + let buf = async_buffer.get_buffer_for_opt(); + let mut buf = buf.lock().unwrap(); + println!("time get_buffer_for_opt: {}", time_tmp.elapsed().unwrap().as_secs_f32()); + + let time_tmp = SystemTime::now(); + let record = agent.opt(&mut buf); + println!("time opt: {}", time_tmp.elapsed().unwrap().as_secs_f32()); + record + }; if let Some(mut record) = record { + let time_tmp = SystemTime::now(); + opt_steps += 1; opt_steps_ += 1; @@ -352,6 +364,7 @@ where info!("Sends the trained model info to ActorManager"); self.sync(&agent); } + println!("time others: {}", time_tmp.elapsed().unwrap().as_secs_f32()); } } info!("Stopped training loop");