Exemple #1
0
    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)
Exemple #2
0
    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)