Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

caffe2到tensorrt的模型转换不识别MAX_BATCH_SIZE参数 #511

Closed
94Bo opened this issue Aug 30, 2021 · 0 comments
Closed

caffe2到tensorrt的模型转换不识别MAX_BATCH_SIZE参数 #511

94Bo opened this issue Aug 30, 2021 · 0 comments

Comments

@94Bo
Copy link
Contributor

94Bo commented Aug 30, 2021 •
edited

caffe2到tensorrt的模型转换，在使用环境变量传参的情况下，不识别MAX_BATCH_SIZE参数
造成传入None作为batch size

使用小写的max_batch_size作为环境变量则能正确识别。

h5 --> tvm也存在这样的问题
@94Bo 94Bo closed this as completed Sep 3, 2021
XueweiHan21 added a commit that referenced this issue Sep 6, 2021
@XueweiHan21
Fix some bugs in model_compiler
e159b30 
Closes #511.

Signed-off-by: HanXuewei <han.xuewei@zte.com.cn>
@XueweiHan21 XueweiHan21 mentioned this issue Sep 6, 2021
Closed
XueweiHan21 added a commit that referenced this issue Sep 8, 2021
@XueweiHan21
Fix some bugs in model_compiler
75e3d27 
Closes #511.

Signed-off-by: HanXuewei <han.xuewei@zte.com.cn>
bors bot pushed a commit that referenced this issue Sep 8, 2021
@XueweiHan21
Fix some bugs in model_compiler (#514)
c30449d 
Closes #511.

Signed-off-by: HanXuewei <han.xuewei@zte.com.cn>
bors bot pushed a commit that referenced this issue Sep 9, 2021
@XueweiHan21
Fix some bugs in model_compiler (#514)
fc0b1ce 
Closes #511.

Signed-off-by: HanXuewei <han.xuewei@zte.com.cn>
bors bot pushed a commit that referenced this issue Sep 9, 2021
@XueweiHan21 @shiyingjin
Fix some bugs in model_compiler (#514)
adf46b0 
Closes #511.

Signed-off-by: HanXuewei <han.xuewei@zte.com.cn>

Co-authored-by: shiyingjin <syjin_2005@163.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

[Merged by Bors] - Fix some bugs in model_compiler Adlik/Adlik
1 participant
@94Bo