5 days ago
-
3D full finetuning: NDHWC train_graph.json + torch_params.json, structured weight_map to_onnx, 3D runtime; 2D spike peak_mb corrected
6 days ago
-
finetune runtime: note on Blob import (HF serves .js as non-JS MIME)
-
Browser finetuning: manifest finetune block (head; full/tfjs for 2D students), initializer offsets, weight maps, torch_version/exporter/opset, source_repo for 2D students
7 days ago
-
2D students: final 60k-step weights (val F1 0.883 setup_53_2d, 0.902 setup_54_2d)
-
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/
-
Add experimental 2D-slice distilled mito students (setup_53/54, step 20000)
8 days ago
-
README: download links, fp32 vs fp16 WebGPU speeds
9 days ago
-
Model card: both graph layouts + browser WebGPU benchmark (Apple GPU, real EM)
-
Restore native-Conv3d graphs under conv3d/ (same models; root setup_*/ hold the faster Conv2d rewrite)
-
All models: export Conv3d as exact Conv2d over depth-folded batch (~16x faster on onnxruntime-web WebGPU); drop experimental/
-
experimental: Conv3d-as-Conv2d (depth-folded batch) variants for WebGPU speed testing
-
Unet3dIso: export 3D ConvTranspose as exact 1x1x1 Conv + depth-to-space (WebGPU EP has no 3D ConvTranspose)
-
Add exp_mito ONNX web models (6 setups x 3 block sizes x fp32/fp16/int8w/int8w_fp16)
-
initial commit