add IJCAI 2024 CAR competition to inedx.md and fix docstrings in cubo…

…id_transformer.py(test=document_fix) (PaddlePaddle#874)
zhaoshuishentiha0 · May 2, 2024 · d4156e2 · d4156e2
1 parent 22329ee
commit d4156e2
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 11 deletions.
diff --git a/docs/index.md b/docs/index.md
@@ -4,6 +4,8 @@
 ./README.md:status
 --8<--
 
+🔥 [IJCAI 2024: 任意三维几何外形车辆的风阻快速预测竞赛](https://competition.atomgit.com/competitionInfo?id=7f3f276465e9e845fd3a811d2d6925b5)，欢迎报名参赛。
+
 <style>
     .container {
         display: flex;

diff --git a/ppsci/arch/cuboid_transformer.py b/ppsci/arch/cuboid_transformer.py
@@ -104,10 +104,10 @@ def forward(self, x):
         """x --> [K x Conv2D] --> PatchMerge
 
         Args:
-            x : (B, T, H, W, C)
+            x: (B, T, H, W, C)
 
         Returns:
-            out : (B, T, H_new, W_new, C_out)
+            out: (B, T, H_new, W_new, C_out)
         """
 
         B, T, H, W, C = x.shape
@@ -178,10 +178,10 @@ def forward(self, x):
         """x --> Upsample --> [K x Conv2D]
 
         Args:
-            x : (B, T, H, W, C)
+            x: (B, T, H, W, C)
 
         Returns:
-            out : (B, T, H_new, W_new, C)
+            out: (B, T, H_new, W_new, C)
         """
 
         x = self.upsample(x)
@@ -286,10 +286,10 @@ def forward(self, x):
         """x --> [K x Conv2D] --> PatchMerge --> ... --> [K x Conv2D] --> PatchMerge
 
         Args:
-            x : (B, T, H, W, C)
+            x: (B, T, H, W, C)
 
         Returns:
-            out : (B, T, H_new, W_new, C_out)
+            out: (B, T, H_new, W_new, C_out)
         """
 
         for i, (conv_block, patch_merge) in enumerate(
@@ -400,10 +400,10 @@ def forward(self, x):
         """x --> Upsample --> [K x Conv2D] --> ... --> Upsample --> [K x Conv2D]
 
         Args:
-            x : Shape (B, T, H, W, C)
+            x: Shape (B, T, H, W, C)
 
         Returns:
-            out : Shape (B, T, H_new, W_new, C)
+            out: Shape (B, T, H_new, W_new, C)
         """
         for i, (conv_block, upsample) in enumerate(
             zip(self.conv_block_list, self.upsample_list)
@@ -915,10 +915,10 @@ def get_initial_z(self, final_mem, T_out):
     def forward(self, x, verbose=False):
         """
         Args:
-            x : Shape (B, T, H, W, C)
-            verbos : if True, print intermediate shapes
+            x: Shape (B, T, H, W, C)
+            verbose: if True, print intermediate shapes
         Returns:
-            out : The output Shape (B, T_out, H, W, C_out)
+            out: The output Shape (B, T_out, H, W, C_out)
         """
 
         x = self.concat_to_tensor(x, self.input_keys)