diff --git a/selfdrive/modeld/modeld.py b/selfdrive/modeld/modeld.py index 3d6341dc29..8e26b78e27 100755 --- a/selfdrive/modeld/modeld.py +++ b/selfdrive/modeld/modeld.py @@ -1,7 +1,7 @@ #!/usr/bin/env python3 from openpilot.system.hardware import TICI -from openpilot.selfdrive.modeld.runners.model_runner import ONNXRunner, TinyGradRunner +from openpilot.selfdrive.modeld.runners.model_runner import ONNXRunner, TinygradRunner # import time @@ -57,7 +57,7 @@ class ModelState: } # Initialize model runner - self.model_runner = TinyGradRunner() if TICI else ONNXRunner(self.frames) + self.model_runner = TinygradRunner() if TICI else ONNXRunner(self.frames) self.parser = Parser() net_output_size = self.model_runner.model_metadata['output_shapes']['outputs'][1] diff --git a/selfdrive/modeld/runners/model_runner.py b/selfdrive/modeld/runners/model_runner.py index 77d1c97ba1..53da7cfd57 100644 --- a/selfdrive/modeld/runners/model_runner.py +++ b/selfdrive/modeld/runners/model_runner.py @@ -49,12 +49,12 @@ class ModelRunner(ABC): return parsed_outputs -class TinyGradRunner(ModelRunner): - """TinyGrad implementation of model runner for TICI hardware.""" +class TinygradRunner(ModelRunner): + """Tinygrad implementation of model runner for TICI hardware.""" def __init__(self): super().__init__() - # Load TinyGrad model + # Load Tinygrad model with open(MODEL_PKL_PATH, "rb") as f: self.model_run = pickle.load(f)