diff --git a/README.md b/README.md
index 43b2003..0557aa2 100644
--- a/README.md
+++ b/README.md
@@ -2,7 +2,7 @@
 
 Support of inference with TensorRT for [sshaoshuai/MTR](https://github.com/sshaoshuai/MTR).
 
-## Network
+## Inputs/Outputs
 
 ### Inputs
 
@@ -28,8 +28,8 @@ where,
 
 ### Outputs
 
-- `scores <float: [N, M]>`
-- `trajectory <float: [N, M, Tf, Dt]>`
+- `scores <float: [B, M]>`
+- `trajectory <float: [B, M, Tf, Dt]>`
 
 where,