|
|
@ -33,8 +33,9 @@ tinygrad_files = ["#"+x for x in glob.glob(env.Dir("#tinygrad_repo").relpath + " |
|
|
|
|
|
|
|
|
|
|
|
# Get model metadata |
|
|
|
# Get model metadata |
|
|
|
fn = File("models/supercombo").abspath |
|
|
|
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' |
|
|
|
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 |
|
|
|
# Compile tinygrad model |
|
|
|
pythonpath_string = 'PYTHONPATH="${PYTHONPATH}:' + env.Dir("#tinygrad_repo").abspath + '"' |
|
|
|
pythonpath_string = 'PYTHONPATH="${PYTHONPATH}:' + env.Dir("#tinygrad_repo").abspath + '"' |
|
|
|