fix initialization

This commit is contained in:
Jason Wen
2025-03-16 15:51:39 -04:00
parent 8a4b3659a0
commit 45e259f438
2 changed files with 5 additions and 5 deletions
+1 -2
View File
@@ -34,8 +34,7 @@ class LatControlTorque(LatControl):
self.use_steering_angle = self.torque_params.useSteeringAngle
self.steering_angle_deadzone_deg = self.torque_params.steeringAngleDeadzoneDeg
if CP_SP.neuralNetworkLateralControl.enabled:
self.nnlc = NeuralNetworkLateralControl(self, CP, CP_SP)
self.nnlc = NeuralNetworkLateralControl(self, CP, CP_SP)
def update_live_torque_params(self, latAccelFactor, latAccelOffset, friction):
self.torque_params.latAccelFactor = latAccelFactor
@@ -80,10 +80,13 @@ class NeuralNetworkLateralControl:
self.lac_torque = lac_torque
self.params = Params()
self.enabled = CP_SP.neuralNetworkLateralControl.enabled
# NN model takes current v_ego, lateral_accel, lat accel/jerk error, roll, and past/future/planned data
# of lat accel and roll
# Past value is computed using previous desired lat accel and observed roll
self.flux_model = FluxModel(CP_SP.neuralNetworkLateralControl.modelPath)
# Only initialize FluxModel if enabled
self.flux_model = FluxModel(CP_SP.neuralNetworkLateralControl.modelPath) if self.enabled else None
self.torque_from_lateral_accel = lac_torque.torque_from_lateral_accel
self.torque_params = lac_torque.torque_params
@@ -102,8 +105,6 @@ class NeuralNetworkLateralControl:
self._pid_log = None
# twilsonco's Lateral Neural Network Feedforward
self.enabled = CP_SP.neuralNetworkLateralControl.enabled
# Instantaneous lateral jerk changes very rapidly, making it not useful on its own,
# however, we can "look ahead" to the future planned lateral jerk in order to gauge
# whether the current desired lateral jerk will persist into the future, i.e.