Load weights and create model
(self)
| 735 | ) |
| 736 | |
| 737 | def load_model(self) -> None: |
| 738 | """Load weights and create model""" |
| 739 | |
| 740 | self.worker.load_model() |
| 741 | loaded_model_signal_data = np.zeros(shape=[1], dtype=np.int32) |
| 742 | self.loaded_model_signal = IPCSignal( |
| 743 | name="loaded_model_signal", |
| 744 | array=loaded_model_signal_data, |
| 745 | dtype=np.int32, |
| 746 | suffix=self.parallel_config.engine_pid, |
| 747 | create=False, |
| 748 | ) |
| 749 | if self.ranks > 1: |
| 750 | paddle.distributed.barrier() |
| 751 | self.loaded_model_signal.value[0] = 1 |
| 752 | |
| 753 | def run_control_method(self, control_request: ControlRequest) -> None: |
| 754 | logger.info(f"Rank: {self.local_rank} Start to run control request: {control_request}") |
no test coverage detected