measure get_buffer_for_opt

laboroai · Oct 1, 2023 · 8aa4bb4 · 8aa4bb4
1 parent 45d1a7a
commit 8aa4bb4
Showing 1 changed file with 14 additions and 1 deletion.
diff --git 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");