From 80f23a8581b5f19ab131f4557c99154bd03affbe Mon Sep 17 00:00:00 2001 From: WJJ1995 Date: Wed, 25 Aug 2021 21:55:38 +0800 Subject: [PATCH] fixed MaxUnpool bug (#662) --- .../API_docs/nn/torch.nn.MaxUnpool1d.md | 5 ++--- .../API_docs/nn/torch.nn.MaxUnpool2d.md | 6 ++---- .../API_docs/nn/torch.nn.MaxUnpool3d.md | 5 ++--- 3 files changed, 6 insertions(+), 10 deletions(-) diff --git a/docs/pytorch_project_convertor/API_docs/nn/torch.nn.MaxUnpool1d.md b/docs/pytorch_project_convertor/API_docs/nn/torch.nn.MaxUnpool1d.md index 53f0d92..e7b876e 100644 --- a/docs/pytorch_project_convertor/API_docs/nn/torch.nn.MaxUnpool1d.md +++ b/docs/pytorch_project_convertor/API_docs/nn/torch.nn.MaxUnpool1d.md @@ -8,10 +8,9 @@ torch.nn.MaxUnpool1d(kernel_size, stride=None, padding=0) ```python import paddle import paddle.nn as nn -TYPE_MAPPER = {"fp16": "float16", "fp32": "float32", "fp64": "float64"} # 定义MaxUnpool1D -class MaxUnpool1D(paddle.nn.Layer): +class MaxUnpool1D(nn.Layer): def __init__(self, kernel_size, stride=None, padding=0): super().__init__() if isinstance(stride, int): @@ -49,7 +48,7 @@ class MaxUnpool1D(paddle.nn.Layer): flatten_indices = paddle.flatten(indices) flatten_input = paddle.flatten(input) for i in range(flatten_indices.shape[0]): - flatten_out[flatten_indices[i].tolist()] = flatten_input[i].tolist() + flatten_out[int(flatten_indices[i])] = flatten_input[i] out = paddle.reshape(flatten_out, out.shape) return out diff --git a/docs/pytorch_project_convertor/API_docs/nn/torch.nn.MaxUnpool2d.md b/docs/pytorch_project_convertor/API_docs/nn/torch.nn.MaxUnpool2d.md index 5efb8ec..fcda1b6 100644 --- a/docs/pytorch_project_convertor/API_docs/nn/torch.nn.MaxUnpool2d.md +++ b/docs/pytorch_project_convertor/API_docs/nn/torch.nn.MaxUnpool2d.md @@ -9,10 +9,9 @@ torch.nn.MaxUnpool2d(kernel_size, stride=None, padding=0) ```python import paddle import paddle.nn as nn -TYPE_MAPPER = {"fp16": "float16", "fp32": "float32", "fp64": "float64"} # 定义MaxUnpool2D -class MaxUnpool2D(paddle.nn.Layer): +class MaxUnpool2D(nn.Layer): def __init__(self, kernel_size, stride=None, padding=0): super().__init__() if isinstance(stride, int): @@ -41,7 +40,6 @@ class MaxUnpool2D(paddle.nn.Layer): if len(output_size) == len(self.kernel_size) + 2: output_size = output_size[2:] t = str(input.dtype).lower().strip().split(".")[-1] - t = TYPE_MAPPER[t] out = paddle.zeros(output_size, dtype=t) flatten_out = paddle.flatten(out) for i in range(indices.shape[0]): @@ -53,7 +51,7 @@ class MaxUnpool2D(paddle.nn.Layer): flatten_indices = paddle.flatten(indices) flatten_input = paddle.flatten(input) for i in range(flatten_indices.shape[0]): - flatten_out[flatten_indices[i].tolist()] = flatten_input[i].tolist() + flatten_out[int(flatten_indices[i])] = flatten_input[i] out = paddle.reshape(flatten_out, out.shape) return out diff --git a/docs/pytorch_project_convertor/API_docs/nn/torch.nn.MaxUnpool3d.md b/docs/pytorch_project_convertor/API_docs/nn/torch.nn.MaxUnpool3d.md index 8e4e778..8084ea9 100644 --- a/docs/pytorch_project_convertor/API_docs/nn/torch.nn.MaxUnpool3d.md +++ b/docs/pytorch_project_convertor/API_docs/nn/torch.nn.MaxUnpool3d.md @@ -8,10 +8,9 @@ torch.nn.MaxUnpool3d(kernel_size, stride=None, padding=0) ```python import paddle import paddle.nn as nn -TYPE_MAPPER = {"fp16": "float16", "fp32": "float32", "fp64": "float64"} # 定义MaxUnpool3D -class MaxUnpool3D(paddle.nn.Layer): +class MaxUnpool3D(nn.Layer): def __init__(self, kernel_size, stride=None, padding=0): super().__init__() if isinstance(stride, int): @@ -55,7 +54,7 @@ class MaxUnpool3D(paddle.nn.Layer): flatten_indices = paddle.flatten(indices) flatten_input = paddle.flatten(input) for i in range(flatten_indices.shape[0]): - flatten_out[flatten_indices[i].tolist()] = flatten_input[i].tolist() + flatten_out[int(flatten_indices[i])] = flatten_input[i] out = paddle.reshape(flatten_out, out.shape) return out -- GitLab