From 37eda2cab71aa8a3e7ff0d54f4f94862a645f7a2 Mon Sep 17 00:00:00 2001
From: lijianshe02 <48898730+lijianshe02@users.noreply.github.com>
Date: Tue, 24 Nov 2020 16:55:10 +0800
Subject: [PATCH] remove fluid code in psgan (#97)

---
 ppgan/models/generators/makeup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppgan/models/generators/makeup.py b/ppgan/models/generators/makeup.py
index 0532e32..5f80ae4 100644
--- a/ppgan/models/generators/makeup.py
+++ b/ppgan/models/generators/makeup.py
@@ -207,7 +207,7 @@ class TNetDown(paddle.nn.Layer):
         return out
 
 
-class GetMatrix(paddle.fluid.dygraph.Layer):
+class GetMatrix(paddle.nn.Layer):
     def __init__(self, dim_in, dim_out):
         super(GetMatrix, self).__init__()
         self.get_gamma = nn.Conv2D(dim_in,
-- 
GitLab