diff --git a/ppgan/models/lapstyle_model.py b/ppgan/models/lapstyle_model.py
index bf090ea46..b3a644e8d 100644
--- 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)