Skip to content

P
PaddleGAN

PaddlePaddle / PaddleGAN
大约 1 年 前同步成功

通知 97
Star 7254
Fork 1210
P
PaddleGAN
未验证 提交 f210ea36 编写于 作者: W wangna11BD 提交者： GitHub
浏览文件
操作

fix lapstyle (#551)

上级 67d7e415
显示空白变更内容
内联 并排
Showing with 2 addition and 2 deletion
ppgan/models/lapstyle_model.py
浏览文件 @ f210ea36
......@@ -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.
先完成此消息的编辑！
想要评论请 注册 或