Change timing

8ff · Sep 1, 2023 · 985880e · 985880e
1 parent 504adc6
commit 985880e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/objectPredict/example/example.go b/pkg/objectPredict/example/example.go
@@ -29,14 +29,14 @@ func bench(config objectPredict.Config, num int) {
 	stats := prettyTimer.NewTimingStats()
 
 	for i := 0; i < num; i++ {
-		start := time.Now()
+		stats.Start()
 		_, _, err = obj.Predict(img)
 		if err != nil {
 			fmt.Println("Cannot predict:", err)
 			return
 		}
 		// End timer
-		stats.RecordTiming(time.Since(start))
+		stats.Finish()
 	}
 	stats.PrintStats()
 }