AXERA-TECH
/

bevformer

Object Detection

Autonomous-Driving

Model card Files Files and versions

fangmingguo commited on 13 days ago

Commit

d92026b

·

verified ·

1 Parent(s): c8a2d3c

Update inference_axmodel.py

Files changed (1) hide show

inference_axmodel.py +1 -1

inference_axmodel.py CHANGED Viewed

@@ -840,7 +840,7 @@ def main():
     os.makedirs(args.output_dir, exist_ok=True)
     # Load AXModel
-    ax_session = load_axmodel(args.axmodel)
     # Get model parameters from config
     transformer_cfg = config['model']['transformer']

     os.makedirs(args.output_dir, exist_ok=True)
     # Load AXModel
+    ax_session = load_axmodel(args.model)
     # Get model parameters from config
     transformer_cfg = config['model']['transformer']