(self, strategy=None)
| 93 | return fetch_dict |
| 94 | |
| 95 | def create_optimizer(self, strategy=None): |
| 96 | optimizer = paddle.optimizer.Adam( |
| 97 | learning_rate=self.learning_rate, lazy_mode=True) |
| 98 | if strategy != None: |
| 99 | import paddle.distributed.fleet as fleet |
| 100 | optimizer = fleet.distributed_optimizer(optimizer, strategy) |
| 101 | optimizer.minimize(self._cost) |
| 102 | |
| 103 | def infer_net(self, input): |
| 104 | return self.net(input, is_infer=True) |