From 056b87414880e0520bb4560fc40d5b62db9c5175 Mon Sep 17 00:00:00 2001 From: lzzyzlbb <287246233@qq.com> Date: Mon, 19 Jul 2021 20:18:56 +0800 Subject: [PATCH] fix the order of unfold parameters (#34156) * fix the order of unfold parameters --- python/paddle/nn/layer/common.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/python/paddle/nn/layer/common.py b/python/paddle/nn/layer/common.py index f608f20fee..1d7f7c6589 100644 --- a/python/paddle/nn/layer/common.py +++ b/python/paddle/nn/layer/common.py @@ -1428,8 +1428,13 @@ class Unfold(layers.Layer): self.name = name def forward(self, input): - return F.unfold(input, self.kernel_sizes, self.dilations, self.paddings, - self.strides, self.name) + return F.unfold( + input, + kernel_sizes=self.kernel_sizes, + strides=self.strides, + paddings=self.paddings, + dilations=self.dilations, + name=self.name) def extra_repr(self): name_str = ', name={}'.format(self.name) if self.name else '' -- GitLab