(self)
| 323 | fishmask_plugin_on_end(self) |
| 324 | |
| 325 | def on_test_start(self): |
| 326 | # Evaluate model on best metric if training set exists |
| 327 | model_fname = os.path.join(self.config.exp_dir, f"global_step{self.best_eval_global_step}.pt") |
| 328 | print(f"Tested on best model step {self.best_eval_global_step} (global_step{self.best_eval_global_step}.pt)") |
| 329 | self.config.load_weight = model_fname |
| 330 | self.load_model() |
| 331 | |
| 332 | def load_model(self): |
| 333 | if self.config.load_weight != "": |
nothing calls this directly
no test coverage detected