|
|
|
@ -32,7 +32,7 @@ lenvCython.Program('models/commonmodel_pyx.so', 'models/commonmodel_pyx.pyx', LI |
|
|
|
|
tinygrad_files = ["#"+x for x in glob.glob(env.Dir("#tinygrad_repo").relpath + "/**", recursive=True, root_dir=env.Dir("#").abspath) if 'pycache' not in x] |
|
|
|
|
|
|
|
|
|
# Get model metadata |
|
|
|
|
for model_name in ['driving_vision', 'driving_policy']: |
|
|
|
|
for model_name in ['driving_vision', 'driving_policy', 'driving_misc']: |
|
|
|
|
fn = File(f"models/{model_name}").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' |
|
|
|
@ -47,7 +47,7 @@ elif arch == 'Darwin': |
|
|
|
|
else: |
|
|
|
|
device_string = 'LLVM=1 LLVMOPT=1 BEAM=0 IMAGE=0' |
|
|
|
|
|
|
|
|
|
for model_name in ['driving_vision', 'driving_policy', 'dmonitoring_model']: |
|
|
|
|
for model_name in ['driving_vision', 'driving_policy', 'driving_misc', 'dmonitoring_model']: |
|
|
|
|
fn = File(f"models/{model_name}").abspath |
|
|
|
|
cmd = f'{pythonpath_string} {device_string} python3 {Dir("#tinygrad_repo").abspath}/examples/openpilot/compile3.py {fn}.onnx {fn}_tinygrad.pkl' |
|
|
|
|
lenv.Command(fn + "_tinygrad.pkl", [fn + ".onnx"] + tinygrad_files, cmd) |
|
|
|
|