pytorch nn 模块中的错误语法无效

Question

我的 pytorch 代码中有错误，我真的不明白为什么。

  File "<ipython-input-11-89006c750b74>", line 3
    def GaussianBlur(torch.nn.Module):
                          ^
SyntaxError: invalid syntax

代码的rest在这里

将 torch.nn 导入为 nn

def GaussianBlur(torch.nn.Module):
    def __init__(self, kernel_size, std_dev):
        self.kernel_size = kernel_size
        self.std_dev = std_dev

    def forward(self, img):
        image = np.array(img)
        image_blur = cv2.GaussianBlur(image, self.kernel_size, self.std_dev)
        return Image.fromarray(image_blur)

有谁知道发生了什么？ 谢谢

Answer 1

def用于函数， class用于类。 这应该有效：

class GaussianBlur(torch.nn.Module):