Commit History

5 days ago

  1. 3D full finetuning: NDHWC train_graph.json + torch_params.json, structured weight_map to_onnx, 3D runtime; 2D spike peak_mb corrected

    Verified signature
    3331842

6 days ago

  1. finetune runtime: note on Blob import (HF serves .js as non-JS MIME)

    Verified signature
    fb68001
  2. Browser finetuning: manifest finetune block (head; full/tfjs for 2D students), initializer offsets, weight maps, torch_version/exporter/opset, source_repo for 2D students

    Verified signature
    1a480eb

7 days ago

  1. 2D students: final 60k-step weights (val F1 0.883 setup_53_2d, 0.902 setup_54_2d)

    Verified signature
    263005f
  2. 2D students to top-level setup_53_2d/ setup_54_2d/ (3D manifest schema, per-axis sizes, step 20000); source_repo on 3D manifests; remove 2d/

    Verified signature
    2c17b0b
  3. Add experimental 2D-slice distilled mito students (setup_53/54, step 20000)

    Verified signature
    1fb3b1b

8 days ago

  1. README: download links, fp32 vs fp16 WebGPU speeds

    Verified signature
    7179352

9 days ago

  1. Model card: both graph layouts + browser WebGPU benchmark (Apple GPU, real EM)

    Verified signature
    e743c69
  2. Restore native-Conv3d graphs under conv3d/ (same models; root setup_*/ hold the faster Conv2d rewrite)

    Verified signature
    665be62
  3. All models: export Conv3d as exact Conv2d over depth-folded batch (~16x faster on onnxruntime-web WebGPU); drop experimental/

    Verified signature
    74ccc9b
  4. experimental: Conv3d-as-Conv2d (depth-folded batch) variants for WebGPU speed testing

    Verified signature
    9aa840f
  5. Unet3dIso: export 3D ConvTranspose as exact 1x1x1 Conv + depth-to-space (WebGPU EP has no 3D ConvTranspose)

    Verified signature
    6a5289f
  6. Add exp_mito ONNX web models (6 setups x 3 block sizes x fp32/fp16/int8w/int8w_fp16)

    Verified signature
    5c02968
  7. initial commit

    Verified signature
    e23e403