Method switch_engine

scripts/trt.py:86–89

(self)

        self.refitted_keys = set(refit_dict.keys())

    def switch_engine(self):
        self.loaded_config = self.configs[self.profile_idx]
        self.engine.reset(os.path.join(TRT_MODEL_DIR, self.loaded_config["filepath"]))
        self.activate()

    def activate(self):
        self.loaded_config = self.configs[self.profile_idx]