(self)
| 283 | self.exe.run(self.startup_program) |
| 284 | |
| 285 | def callback_train(self): |
| 286 | epoch_start_time = time.time() |
| 287 | self.set_dump_fields() |
| 288 | fetch_info = [ |
| 289 | f"Epoch {self.epoch_idx} Var {var_name}" |
| 290 | for var_name in self.metrics |
| 291 | ] |
| 292 | self.exe.train_from_dataset( |
| 293 | program=self.main_program, |
| 294 | dataset=self.train_dataset, |
| 295 | fetch_list=self.fetch_vars, |
| 296 | fetch_info=fetch_info, |
| 297 | print_period=self.print_step, |
| 298 | debug=self.debug, |
| 299 | ) |
| 300 | self.epoch_idx += 1 |
| 301 | epoch_time = time.time() - epoch_start_time |
| 302 | epoch_speed = self.train_example_nums / epoch_time |
| 303 | self.train_statical_info["speed"].append(epoch_speed) |
| 304 | logger.info("fl-ps > callback_train finished") |
| 305 | |
| 306 | def callback_infer(self): |
| 307 | fetch_info = [ |
nothing calls this directly
no test coverage detected