Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/tilelang.git" did not exist on "782ca9f64ee7897eecc7cacb40a5fdb9838ea270"
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
Hide 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