Support multiple kmodels for accuracy_test.

kendryte · Jan 29, 2024 · fa0da60 · fa0da60
1 parent de6c89d
commit fa0da60
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/accuracy_test/nuc_proxy_accuracy_test.py b/tests/accuracy_test/nuc_proxy_accuracy_test.py
@@ -176,8 +176,12 @@ def recv_worker(conn, target):
     if dict['app'] == 1:
         dict['app'] = recv_file(conn, case_dir, target.logger)
 
-    if dict['kmodel'] == 1:
-        dict['kmodel'] = recv_file(conn, case_dir, target.logger)
+    kmodel_num = dict['kmodel']
+    if kmodel_num > 0:
+        file_names = []
+        for i in range(kmodel_num):
+            file_names.append(recv_file(conn, case_dir, target.logger))
+        dict['kmodel'] = ' '.join(file_names)
 
     if dict['description'] == 1:
         dict['description'] = recv_file(conn, case_dir, target.logger)