Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "c8d1ed9d41756c764229ee531d65f776191f547a"
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):
kernel_size=convtrans.kernel_size,
stride=convtrans.stride,
padding=convtrans.padding,
output_padding=convtrans.output_padding,
dilation=convtrans.dilation,
groups=new_groups,
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