def run_inference(self): if not hasattr(self.opt_model, 'name'): logger.info("no 'update_optimal_model' method, checking for model.txt file . . . ") self.update_optimal_model() gun = Launcher(self.opt_model) gun.predict(self.path_to_best_checkpoint)
def run_inference(self): if not hasattr(self.opt_model, 'name'): logger.info("no 'update_optimal_model' method, checking for model.txt file . . . ") self.update_optimal_model() gun = Launcher(self.opt_model, remote=self.remote) path_to_first_checkpoint = self.path_to_best_checkpoint.split('.')[0] + str(0) + '.pt' gun.predict(path_to_first_checkpoint)