mawei/onepilot
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

modeld: trigger model metadata on script change (#34623)

Browse Source
This commit is contained in:
YassineYousfi
2025-02-19 15:11:33 -08:00
committed by GitHub
parent f3004c22ce
commit 9ec69ddb36
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
selfdrive/modeld/SConscript
View File

@@ -33,8 +33,9 @@ tinygrad_files = ["#"+x for x in glob.glob(env.Dir("#tinygrad_repo").relpath + "
# Get model metadata
fn = File("models/supercombo").abspath
script_files = [File(Dir("#selfdrive/modeld").File("get_model_metadata.py").abspath)]
cmd = f'python3 {Dir("#selfdrive/modeld").abspath}/get_model_metadata.py {fn}.onnx'
lenv.Command(fn + "_metadata.pkl", [fn + ".onnx"] + tinygrad_files, cmd)
lenv.Command(fn + "_metadata.pkl", [fn + ".onnx"] + tinygrad_files + script_files, cmd)
# Compile tinygrad model
pythonpath_string = 'PYTHONPATH="${PYTHONPATH}:' + env.Dir("#tinygrad_repo").abspath + '"'