PFLD-pytorch icon indicating copy to clipboard operation
PFLD-pytorch copied to clipboard

推理问题

Open yfding98 opened this issue 1 year ago • 0 comments

再将onnx模型转为动态批次后进行批量推理时报错,查了一下说是gemm_106算子维度不匹配,以batch_size=8为例,我这里的输入是{1024, 32},而你的weight是{32,128}: image

yfding98 avatar Sep 06 '24 03:09 yfding98