[README]

kyegomez · Dec 30, 2023 · f43df2b · f43df2b
1 parent 2992fae
commit f43df2b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -16,18 +16,22 @@ from qformer import QFormer
 x = torch.randn(
     1, 32, 512
 )  # Create a random tensor of shape (1, 32, 512)
+
 img = torch.randn(
     1, 32, 512
 )  # Create another random tensor of shape (1, 32, 512)
 
 qformer = QFormer(
     512, 8, 8, 0.1, 2, 2
 )  # Create an instance of the QFormer model
+
 y = qformer(
     x, img
 )  # Apply the QFormer model to the input tensors x and img
+
 print(y.shape)  # Print the shape of the output tensor y
 
+
 ```
 
 

diff --git a/example.py b/example.py
@@ -4,14 +4,17 @@
 x = torch.randn(
     1, 32, 512
 )  # Create a random tensor of shape (1, 32, 512)
+
 img = torch.randn(
     1, 32, 512
 )  # Create another random tensor of shape (1, 32, 512)
 
 qformer = QFormer(
     512, 8, 8, 0.1, 2, 2
 )  # Create an instance of the QFormer model
+
 y = qformer(
     x, img
 )  # Apply the QFormer model to the input tensors x and img
+
 print(y.shape)  # Print the shape of the output tensor y