Add detail of sample statistics for different model category and call it in ci.

Author: Xreki

.github/workflows/Validate-GPU.yml

@@ -78,12 +78,19 @@ jobs:
       - name: Run check
         env:
           work_dir: ${{ github.workspace }}
+        if: steps.check-bypass.outputs.can-skip != 'true'
         run: |
           docker exec -t ${{ env.container_name }} /bin/bash -c '
           source ${{ github.workspace }}/../../../proxy
           bash ${work_dir}/tools/ci/check_validate.sh
           '
 
+      - name: Sample statistics
+        if: always()
+        run: |
+          set +e
+          python ${{ github.workspace }}/count_sample.py
+
       - name: Terminate and delete the container
         if: always()
         run: |

tools/count_sample.py

@@ -5,10 +5,19 @@
 filename = os.path.abspath(__file__)
 root_dir = os.path.dirname(os.path.dirname(filename))
 samples_dir = os.path.join(root_dir, "samples")
+model_categories = os.listdir(samples_dir)
 
 graph_net_count = 0
-for root, dirs, files in os.walk(samples_dir):
-    for file in files:
-        if file == "graph_net.json":
+graph_net_dict = {}
+for category in model_categories:
+    graph_net_dict[category] = 0
+    category_dir = os.path.join(samples_dir, category)
+    for root, dirs, files in os.walk(category_dir):
+        if "graph_net.json" in files:
             graph_net_count += 1
+            graph_net_dict[category] += 1
+
 print(f"Number of graph_net.json files: {graph_net_count}")
+for name, number in graph_net_dict.items():
+    print(f"- {name:24}: {number}")
+print()