mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-06-15 12:05:01 +08:00
cb32793300
* Off policy model * 2f70b996-c604-4a46-9ac9-13ce7534605b/100 * misc fixes * 1cc1791b-4555-41ce-a5cb-ce046967075a/100 * fix model * 6ab6fae5-fbbd-4ad0-928a-b33794f60dba/100 * recomp * update models * qxfinally correct * b8b96ac6-7918-401a-a862-eaf1fdbba88d/100 * wrong plan * wrong plan * Vf9b3fb5f-4d0d-4dcb-bc3a-5e94d1fdcdaa/200 * bump dbc * ready to merge * rename to on-policy * Just cleanup big models for now --------- Co-authored-by: Kacper Rączy <gfw.kra@gmail.com>
49 lines
1.5 KiB
Python
Executable File
49 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import glob
|
|
|
|
from tinygrad.nn.onnx import OnnxPBParser
|
|
|
|
BASEDIR = os.path.abspath(os.path.join(os.path.dirname(os.path.realpath(__file__)), "../"))
|
|
|
|
MASTER_PATH = os.getenv("MASTER_PATH", BASEDIR)
|
|
MODEL_PATH = "/selfdrive/modeld/models/"
|
|
|
|
|
|
class MetadataOnnxPBParser(OnnxPBParser):
|
|
def _parse_ModelProto(self) -> dict:
|
|
obj = {"metadata_props": []}
|
|
for fid, wire_type in self._parse_message(self.reader.len):
|
|
match fid:
|
|
case 14:
|
|
obj["metadata_props"].append(self._parse_StringStringEntryProto())
|
|
case _:
|
|
self.reader.skip_field(wire_type)
|
|
return obj
|
|
|
|
|
|
def get_checkpoint(f):
|
|
model = MetadataOnnxPBParser(f).parse()
|
|
metadata = {prop["key"]: prop["value"] for prop in model["metadata_props"]}
|
|
return metadata['model_checkpoint'].split('/')[0]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("| | master | PR branch |")
|
|
print("|-| ----- | --------- |")
|
|
|
|
for f in glob.glob(BASEDIR + MODEL_PATH + "/*.onnx"):
|
|
# TODO: add checkpoint to DM
|
|
if "dmonitoring" in f:
|
|
continue
|
|
|
|
fn = os.path.basename(f)
|
|
master_path = MASTER_PATH + MODEL_PATH + fn
|
|
if os.path.exists(master_path):
|
|
master = get_checkpoint(master_path)
|
|
master_col = f"[{master}](https://reporter.comma.life/experiment/{master})"
|
|
else:
|
|
master_col = "N/A (new model)"
|
|
pr = get_checkpoint(BASEDIR + MODEL_PATH + fn)
|
|
print("|", fn, "|", master_col, "|", f"[{pr}](https://reporter.comma.life/experiment/{pr})", "|")
|