Skip to content

Commit f210ea3

Browse files
authored
fix lapstyle (#551)
1 parent 67d7e41 commit f210ea3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ppgan/models/lapstyle_model.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -186,7 +186,7 @@ def __init__(self,
186186
self.nets['net_dec'] = build_generator(draftnet_decode)
187187

188188
self.set_requires_grad([self.nets['net_enc']], False)
189-
self.set_requires_grad([self.nets['net_enc']], False)
189+
self.set_requires_grad([self.nets['net_dec']], False)
190190

191191
# define revision-net params
192192
self.nets['net_rev'] = build_generator(revnet_generator)
@@ -322,7 +322,7 @@ def __init__(self,
322322
self.nets['net_enc'] = build_generator(draftnet_encode)
323323
self.nets['net_dec'] = build_generator(draftnet_decode)
324324
self.set_requires_grad([self.nets['net_enc']], False)
325-
self.set_requires_grad([self.nets['net_enc']], False)
325+
self.set_requires_grad([self.nets['net_dec']], False)
326326

327327
# define the first revnet params
328328
self.nets['net_rev'] = build_generator(revnet_generator)

0 commit comments

Comments
 (0)