mawei/sunnypilot
1
0
Fork 0
mirror of https://github.com/sunnypilot/sunnypilot.git synced 2026-08-04 15:46:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

lint be crazy now, man i've been away a while

Browse Source
This commit is contained in:
discountchubbs
2026-08-01 15:15:54 -07:00
parent d086f83aed
commit c22f58dae3
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
openpilot/sunnypilot/modeld_v2/compile_modeld.py
+2 -2
View File
@@ -179,11 +179,11 @@ def compile_and_warmup(nv12: NV12Frame, model_size: tuple[int, int], prepare_onl
queues, npy_arrays = generate_queues_and_npy(all_shapes, frame_skip, Device.DEFAULT)
for i in range(3):
np.random.default_rng(seed=(42 + i))
rng = np.random.default_rng(42 + i)
frame = Tensor.randint(nv12.size, low=0, high=256, dtype=dtypes.uint8, device=WARP_DEV).realize()
big_frame = Tensor.randint(nv12.size, low=0, high=256, dtype=dtypes.uint8, device=WARP_DEV).realize()
for arr in npy_arrays.values():
arr[:] = np.random.randn(*arr.shape).astype(arr.dtype)
arr[:] = rng.standard_normal(arr.shape).astype(arr.dtype)
Device.default.synchronize()
start_time = time.perf_counter()
openpilot/sunnypilot/modeld_v2/parse_model_outputs_split.py
+1
View File
@@ -65,6 +65,7 @@ class Parser:
weights[fidx] = weights[fidx][idxs]
pred_mu[fidx] = pred_mu[fidx][idxs]
pred_std[fidx] = pred_std[fidx][idxs]
assert out_shape is not None
full_shape = tuple([raw.shape[0], in_N] + list(out_shape))
outs[name + '_weights'] = weights
outs[name + '_hypotheses'] = pred_mu.reshape(full_shape)