(self, lr_cfg, optimizer)
| 104 | self.nets['discriminator'] = self.nets['Dis'].discriminator |
| 105 | |
| 106 | def setup_optimizers(self, lr_cfg, optimizer): |
| 107 | self.setup_net_parallel() |
| 108 | # init params |
| 109 | init_weight(self.nets['kp_detector']) |
| 110 | init_weight(self.nets['generator']) |
| 111 | init_weight(self.nets['discriminator']) |
| 112 | |
| 113 | # define loss functions |
| 114 | self.losses = {} |
| 115 | |
| 116 | self.optimizers['optimizer_KP'] = build_optimizer( |
| 117 | optimizer, |
| 118 | self.kp_lr, |
| 119 | parameters=self.nets['kp_detector'].parameters()) |
| 120 | self.optimizers['optimizer_Gen'] = build_optimizer( |
| 121 | optimizer, |
| 122 | self.gen_lr, |
| 123 | parameters=self.nets['generator'].parameters()) |
| 124 | self.optimizers['optimizer_Dis'] = build_optimizer( |
| 125 | optimizer, |
| 126 | self.dis_lr, |
| 127 | parameters=self.nets['discriminator'].parameters()) |
| 128 | |
| 129 | def setup_input(self, input): |
| 130 | self.input_data = input |
no test coverage detected