Hi,
Would you mind the check this script?
import onnx_graphsurgeon as gs
import onnx
batch = 2
graph = gs.import_onnx(onnx.load("arcface_mobile.onnx"))
for inp in graph.inputs:
inp.shape[0] = batch
for out in graph.outputs:
out.shape[0] = batch
onnx.save(gs.export_onnx(graph), "arcface_mobile_bs.onnx")
Thanks.