多机并行计算

2025-08-14 10:09:54 +08:00
parent bc896fc688
commit 99a204ee22
18 changed files with 105 additions and 55 deletions
--- a/tools/model_rknn_transform.py
+++ b/tools/model_rknn_transform.py
@ -122,8 +122,9 @@ if __name__ == '__main__':

    # Build model
    print('--> Building model')
-    ret = rknn.build(do_quantization=False,  # True
-                     dataset='./dataset.txt',
+    ret = rknn.build(do_quantization=True,  # True
+                     # dataset='./dataset.txt',
+                     dataset=conf['base']['dataset'],
                     rknn_batch_size=conf['models']['rknn_batch_size'])
    # ret = rknn.build(do_quantization=False, dataset='./dataset.txt')
    if ret != 0: