mirror of
https://github.com/sunnypilot/sunnypilot.git
synced 2026-08-05 13:25:40 +08:00
Rename TinyGradRunner to TinygradRunner for consistency.
The class name and references were updated to maintain naming consistency across the codebase. This aligns with naming conventions and improves code clarity. No functional changes were introduced.
This commit is contained in:
@@ -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]
|
||||
|
||||
@@ -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)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user