未验证 提交 f210ea36 编写于 作者: W wangna11BD 提交者: GitHub

fix lapstyle (#551)

上级 67d7e415
......@@ -186,7 +186,7 @@ class LapStyleRevFirstModel(BaseModel):
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 @@ class LapStyleRevSecondModel(BaseModel):
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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册