Update fn name

EricLBuehler · Mar 12, 2024 · c3ef7cf · c3ef7cf
1 parent 58a37cd
commit c3ef7cf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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
     }
 }