Skip to content

GitLab

"...composable_kernel_onnxruntime.git" did not exist on "5f82fdd9d30e7d520edf742eb7fb8e5c22c0d78f"
Unverified Commit 65fad189 authored by haoshuai-orka's avatar haoshuai-orka Committed by GitHub
Browse files

supported output_padding for convtranspose2d in model speedup (#4827)

parent 7003603c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
nni/compression/pytorch/speedup/compress_modules.py
View file @ 65fad189
...@@ -512,6 +512,7 @@ def replace_convtranspose2d(convtrans, masks): ...@@ -512,6 +512,7 @@ def replace_convtranspose2d(convtrans, masks):
kernel_size=convtrans.kernel_size, kernel_size=convtrans.kernel_size,
stride=convtrans.stride, stride=convtrans.stride,
padding=convtrans.padding, padding=convtrans.padding,
output_padding=convtrans.output_padding,
dilation=convtrans.dilation, dilation=convtrans.dilation,
groups=new_groups, groups=new_groups,
bias=convtrans.bias is not None, bias=convtrans.bias is not None,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment