未验证 提交 4d9609d0 编写于 作者: L lzzyzlbb 提交者: GitHub

Fom mobile (#412)

* add fom lite training

* add fom lite training

* modify according to review

* modify according to review

* fix error of fom trian error
上级 a449659e
......@@ -294,14 +294,13 @@ class FirstOrderModelMobile(FirstOrderModel):
self.nets['kp_detector'] = self.nets[
'Gen_Full']._layers.kp_extractor
self.nets['generator'] = self.nets['Gen_Full']._layers.generator
self.kp_detector_ori = self.Gen_Full_ori._layers.kp_extractor
self.nets['generator'] = self.nets['Gen_Full']._layers.generator
self.nets['discriminator'] = self.nets['Dis']._layers.discriminator
else:
self.nets['kp_detector'] = self.nets['Gen_Full'].kp_extractor
self.nets['generator'] = self.nets['Gen_Full'].generator
self.kp_detector_ori = self.Gen_Full_ori.kp_extractor
self.nets['discriminator'] = self.nets['Dis'].discriminator
self.kp_detector_ori = self.Gen_Full_ori.kp_extractor
from ppgan.utils.download import get_path_from_url
vox_cpk_weight_url = 'https://paddlegan.bj.bcebos.com/applications/first_order_model/vox-cpk.pdparams'
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册