Update README.md

README.md

@@ -81,5 +81,5 @@ csi = torch.rand((2, 100, 52))
 time_stamp = torch.rand((2, 100))
 attention_mask = torch.zeros((2, 100))
 y = model(csi,time_stamp,attention_mask)
-print(y.shape) # dim: [2,100,64] (batch_size,length,hidden_dim)
+print(y.shape) # dim: [2,100,128] (batch_size,length,hidden_dim)
 ```