(self)
| 89 | self.activate() |
| 90 | |
| 91 | def activate(self): |
| 92 | self.loaded_config = self.configs[self.profile_idx] |
| 93 | if self.engine is None: |
| 94 | self.engine = Engine( |
| 95 | os.path.join(TRT_MODEL_DIR, self.loaded_config["filepath"]) |
| 96 | ) |
| 97 | self.engine.load() |
| 98 | print(f"\nLoaded Profile: {self.profile_idx}") |
| 99 | print(self.engine) |
| 100 | self.engine_vram_req = self.engine.engine.device_memory_size |
| 101 | self.engine.activate(True) |
| 102 | |
| 103 | def deactivate(self): |
| 104 | del self.engine |
no test coverage detected