diff --git a/candle-nn/src/layer_norm.rs b/candle-nn/src/layer_norm.rs index 594942f169..04fd187b93 100644 --- a/candle-nn/src/layer_norm.rs +++ b/candle-nn/src/layer_norm.rs @@ -140,13 +140,13 @@ impl crate::Module for LayerNorm { let start = SystemTime::now() .duration_since(UNIX_EPOCH) .expect("Time travel has occurred!") - .as_millis(); + .as_micros(); let res = self.forward_slow(x); let end = SystemTime::now() .duration_since(UNIX_EPOCH) .expect("Time travel has occurred!") - .as_millis(); - println!("{}ms", end - start); + .as_micros(); + println!("{}us", end - start); res } }