44 lines
1.3 KiB
Python
Executable File
44 lines
1.3 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 = get_checkpoint(MASTER_PATH + MODEL_PATH + fn)
|
|
pr = get_checkpoint(BASEDIR + MODEL_PATH + fn)
|
|
print("|", fn, "|", f"[{master}](https://reporter.comma.life/experiment/{master})", "|", f"[{pr}](https://reporter.comma.life/experiment/{pr})", "|")
|