fix lapstyle (#551)

PaddlePaddle · Jan 4, 2022 · f210ea3 · f210ea3
1 parent 67d7e41
commit f210ea3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ppgan/models/lapstyle_model.py b/ppgan/models/lapstyle_model.py
@@ -186,7 +186,7 @@ def __init__(self,
         self.nets['net_dec'] = build_generator(draftnet_decode)
 
         self.set_requires_grad([self.nets['net_enc']], False)
-        self.set_requires_grad([self.nets['net_enc']], False)
+        self.set_requires_grad([self.nets['net_dec']], False)
 
         # define revision-net params
         self.nets['net_rev'] = build_generator(revnet_generator)
@@ -322,7 +322,7 @@ def __init__(self,
         self.nets['net_enc'] = build_generator(draftnet_encode)
         self.nets['net_dec'] = build_generator(draftnet_decode)
         self.set_requires_grad([self.nets['net_enc']], False)
-        self.set_requires_grad([self.nets['net_enc']], False)
+        self.set_requires_grad([self.nets['net_dec']], False)
 
         # define the first revnet params
         self.nets['net_rev'] = build_generator(revnet_generator)