(self)
| 93 | } |
| 94 | |
| 95 | def setup_net_parallel(self): |
| 96 | if isinstance(self.nets['Gen_Full'], paddle.DataParallel): |
| 97 | self.nets['kp_detector'] = self.nets[ |
| 98 | 'Gen_Full']._layers.kp_extractor |
| 99 | self.nets['generator'] = self.nets['Gen_Full']._layers.generator |
| 100 | self.nets['discriminator'] = self.nets['Dis']._layers.discriminator |
| 101 | else: |
| 102 | self.nets['kp_detector'] = self.nets['Gen_Full'].kp_extractor |
| 103 | self.nets['generator'] = self.nets['Gen_Full'].generator |
| 104 | self.nets['discriminator'] = self.nets['Dis'].discriminator |
| 105 | |
| 106 | def setup_optimizers(self, lr_cfg, optimizer): |
| 107 | self.setup_net_parallel() |
no outgoing calls