Focus project on JAX PPO

scripts/export_jax_ppo_onnx.py

@@ -4,6 +4,7 @@
 from __future__ import annotations
 
 import argparse
+import hashlib
 import json
 from pathlib import Path
 
@@ -86,10 +87,14 @@ def export_model(checkpoint: Path, config: Path | None, output: Path) -> None:
     )
     output.parent.mkdir(parents=True, exist_ok=True)
     onnx.save(model, output)
+    model_bytes = output.read_bytes()
     manifest = {
         "format": "coolrl-lost-cities-jax-ppo-onnx-v1",
-        "source_checkpoint": str(checkpoint.resolve()),
-        "source_config": str(config_path.resolve()),
+        "model_file": output.name,
+        "model_size_bytes": len(model_bytes),
+        "model_sha256": hashlib.sha256(model_bytes).hexdigest(),
+        "source_checkpoint": checkpoint.name,
+        "source_config": config_path.name,
         "observation_size": OBS_DIM,
         "action_size": N_ACTIONS,
         "hidden_size": cfg.network.hidden_size,