User-2468 commited on
Commit
4981240
Β·
verified Β·
1 Parent(s): 5421e04

Record final visual decision and retain v3 production model

Browse files
DEPLOYMENT.md CHANGED
@@ -1,7 +1,7 @@
1
  # Deployment and final-run promotion
2
 
3
  The live Space is https://huggingface.co/spaces/User-2468/mini-unet-colorizer.
4
- Default model revision remains `1a9eb8af2754ad2329a24cfe50d388cb559441d0` (v3.0). The final longer job is https://huggingface.co/jobs/User-2468/6abaef6552d0dbd7f1da7286.
5
 
6
  ## ZeroGPU
7
  Use `app.py`, `inference.py`, `semantic_model.py`, `decision_model.py`, and `requirements-space.txt` (renamed `requirements.txt` in the Space). Space README SDK is Gradio6.28.0; requirements use the same version. Select ZeroGPU hardware. Import `spaces` before `torch`; load/place the model on CUDA at startup. Only chroma inference runs inside `@spaces.GPU(duration=15)`. CPU handles image preparation, guided rendering, and cached strength changes. Do not enable `torch.compile` in ZeroGPU.
 
1
  # Deployment and final-run promotion
2
 
3
  The live Space is https://huggingface.co/spaces/User-2468/mini-unet-colorizer.
4
+ Default model revision remains `1a9eb8af2754ad2329a24cfe50d388cb559441d0` (v3.0). The final longer job completed at https://huggingface.co/jobs/User-2468/6abaef6552d0dbd7f1da7286. Its selected checkpoint was rejected for production; see reports/FINAL_CHECKPOINT_REVIEW.md.
5
 
6
  ## ZeroGPU
7
  Use `app.py`, `inference.py`, `semantic_model.py`, `decision_model.py`, and `requirements-space.txt` (renamed `requirements.txt` in the Space). Space README SDK is Gradio6.28.0; requirements use the same version. Select ZeroGPU hardware. Import `spaces` before `torch`; load/place the model on CUDA at startup. Only chroma inference runs inside `@spaces.GPU(duration=15)`. CPU handles image preparation, guided rendering, and cached strength changes. Do not enable `torch.compile` in ZeroGPU.
README.md CHANGED
@@ -19,9 +19,9 @@ A compact automatic photo colouriser with **3,994,676 learned parameters in tota
19
 
20
  [Try the ZeroGPU Space](https://huggingface.co/spaces/User-2468/mini-unet-colorizer).
21
 
22
- Root weights and ONNX remain the reviewed **v3.0** release. A final continuation of the3,995,832-parameter four-palette challenger is running under [`experiments/final-20260929`](experiments/final-20260929); it is **not yet approved as a production replacement**. See the [visual review and training decision](reports/FINAL_TRAINING_REVIEW.md) and [exact configuration](training/final_config.json). The pilot showed modest patch reductions, not elimination of blotches.
23
 
24
- The updated Space offers Standard, Detailed and Maximum input detail, full-resolution PNG output, and cached strength adjustments. Higher input detail can help some photos but can also change their colours. The default stays at256 pixels. The Space can load future multi-palette checkpoints after review without downloading arbitrary Python code.
25
 
26
  Historical research documents have moved to `reports/history/`. See [the report index](reports/README.md). The `stable` branch and existing release weights are preserved.
27
 
 
19
 
20
  [Try the ZeroGPU Space](https://huggingface.co/spaces/User-2468/mini-unet-colorizer).
21
 
22
+ Root weights and ONNX are the reviewed **v3.0 production release**. The completed 3,995,832-parameter four-palette continuation was **not promoted** after [final checkpoint review](reports/FINAL_CHECKPOINT_REVIEW.md): it lowered patch measures modestly but reduced colour coverage and left historical photos muted. Its selected checkpoint and all training artifacts remain at [experiments/final-20260929](experiments/final-20260929). See the [experiment decision](reports/FINAL_TRAINING_REVIEW.md) and [configuration](training/final_config.json).
23
 
24
+ The updated Space offers Standard, Detailed and Maximum input detail, full-resolution PNG output, and cached strength adjustments. Higher input detail can help some photos but can also change their colours. The default stays at256 pixels. The Space can load compatible multi-palette checkpoints for experiments without downloading arbitrary Python code; its default is v3.
25
 
26
  Historical research documents have moved to `reports/history/`. See [the report index](reports/README.md). The `stable` branch and existing release weights are preserved.
27
 
SHA256SUMS.json CHANGED
@@ -1,8 +1,8 @@
1
  {
2
- "DEPLOYMENT.md": "dac782d3f07838731ced3c0b7a21cfd24ca9943ea9e9660ba6e0dfa5f86de5c0",
3
  "PIPELINE_EVAL.json": "595be88704d580c01a3c4dd642409b3dc5dd25e86ca1c7eb40a696605f181a19",
4
  "QA.json": "b249b1d49045e31b0bf85f9e01515b91a72c4dd973963751c30d14272d262b90",
5
- "README.md": "cdca79dbc42aeb31be15b61adc67f50746096b8fab9d95963232622cc8f691b5",
6
  "RELEASE_MANIFEST.json": "872eebfb322c8990b5e6ac493e4e4068b87f8daac0ddf219b8b55299514ca03c",
7
  "RELEASE_REPORT.md": "69677ec5e13f5c1249f59b01c31ed1b0f1588bddc216bce9167b5d31423a5574",
8
  "app.py": "9fbfae85bc7e9976aee57229cb0825a05b836e31177a4288b2b1efebb075e58b",
@@ -16,6 +16,7 @@
16
  "inference.py": "0ac1f382205b42cef0afaa9173017fa660e6859194a65a61b9b9052c437cd476",
17
  "model.py": "31628744fb5d362d91342063c90eb38b2b97fda8c91e303e4a58b6f388f53c55",
18
  "model.safetensors": "ec1f27d74533adc83f7ab3639a091fc4d8738a434dafc7d172c7873c28a9e715",
 
19
  "reports/FINAL_TRAINING_REVIEW.md": "2c354152ed07c22852401cff66865782840ffb143941a8f92a7ed4fd8542988f",
20
  "reports/README.md": "b50c4af1398d0a8c56bd6bf1a1561f95bd8187c68f057f803ef5fddf4c718f90",
21
  "reports/SPACE_QA_multi.json": "990353b85936624c24a71b4c2183d158251cd122ff23b1ee026eae94bca5715b",
 
1
  {
2
+ "DEPLOYMENT.md": "1ecb94969482959a65e40a6fbd41c985dd3c9df1ff23ad7b13c4f22545c8782b",
3
  "PIPELINE_EVAL.json": "595be88704d580c01a3c4dd642409b3dc5dd25e86ca1c7eb40a696605f181a19",
4
  "QA.json": "b249b1d49045e31b0bf85f9e01515b91a72c4dd973963751c30d14272d262b90",
5
+ "README.md": "39654812ef18de704d72b364d75b3ae8fe45b21751e4b73bcca3418a936578c7",
6
  "RELEASE_MANIFEST.json": "872eebfb322c8990b5e6ac493e4e4068b87f8daac0ddf219b8b55299514ca03c",
7
  "RELEASE_REPORT.md": "69677ec5e13f5c1249f59b01c31ed1b0f1588bddc216bce9167b5d31423a5574",
8
  "app.py": "9fbfae85bc7e9976aee57229cb0825a05b836e31177a4288b2b1efebb075e58b",
 
16
  "inference.py": "0ac1f382205b42cef0afaa9173017fa660e6859194a65a61b9b9052c437cd476",
17
  "model.py": "31628744fb5d362d91342063c90eb38b2b97fda8c91e303e4a58b6f388f53c55",
18
  "model.safetensors": "ec1f27d74533adc83f7ab3639a091fc4d8738a434dafc7d172c7873c28a9e715",
19
+ "reports/FINAL_CHECKPOINT_REVIEW.md": "981b8f369b3950cbd116d865d846b7be59fdae2f21b4c6b7f1eb8c1e74010c02",
20
  "reports/FINAL_TRAINING_REVIEW.md": "2c354152ed07c22852401cff66865782840ffb143941a8f92a7ed4fd8542988f",
21
  "reports/README.md": "b50c4af1398d0a8c56bd6bf1a1561f95bd8187c68f057f803ef5fddf4c718f90",
22
  "reports/SPACE_QA_multi.json": "990353b85936624c24a71b4c2183d158251cd122ff23b1ee026eae94bca5715b",
reports/FINAL_CHECKPOINT_REVIEW.md ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Final checkpoint review β€” 29 September 2026
2
+
3
+ ## Decision
4
+ Keep **v3.0** as the production model. The completed four-palette multi_mix continuation is retained on main at experiments/final-20260929/multi_mix/candidate, but is not promoted to root weights or the Space default. The stable branch is unchanged.
5
+
6
+ The final run completed 16,000 additional updates, then stopped by the prespecified development patience rule. The development-selected candidate was step 4,000 (after the prior 2,000-step pilot), not the last training step. It has 3,995,832 parameters in total including the MobileNetV3 encoder.
7
+
8
+ ## Held-out comparison against v3
9
+ | Measure | Gray v3 β†’ candidate | Film gray v3 β†’ candidate |
10
+ |---|---:|---:|
11
+ | Guided patch excess (lower) | 0.794 β†’ 0.767 | 0.745 β†’ 0.698 |
12
+ | Raw patch excess (lower) | 0.996 β†’ 0.947 | 0.937 β†’ 0.862 |
13
+ | Neutral spill (lower) | 0.234 β†’ 0.197 | 0.245 β†’ 0.216 |
14
+ | Colour coverage (higher) | 51.50% β†’ 48.38% | 50.68% β†’ 47.98% |
15
+ | Mean chroma | 13.24 β†’ 12.54 | 12.95 β†’ 12.33 |
16
+ | Lab ab error (lower) | 12.64 β†’ 12.52 | 12.79 β†’ 12.71 |
17
+
18
+ Exploratory paired 95% intervals for guided patch change are [-0.0619, +0.0049] on gray and [-0.0800, -0.0145] on film. The corresponding coverage drops have intervals [-0.0468, -0.0158] and [-0.0437, -0.0108]. This is not a human preference study.
19
+
20
+ I reviewed all fourteen archival comparisons, held-out modern production grids and special probes. Most archival outputs are nearly identical and remain warm/sepia or muted. The candidate smooths some local artifacts and preserves useful sky/foliage in a few images, but more clothes, interiors and backgrounds remain essentially gray. The Moon probe is still implausibly pink. Neither model eliminates residual blotches. The modest patch improvement does not compensate for the consistent loss of colour on the user's main grayscale-photo use case.
21
+
22
+ The pilot had a more favourable colour-retention trade-off; this continuation moved toward desaturation. Selection within the final course did not establish that it surpasses the released v3. Checkpoints, metrics, source, diagnostics and comparison grids are saved under experiments/final-20260929, with summary.json and multi_mix_test_paired.json providing exact figures.
23
+
24
+ ## Production state
25
+ Root model.safetensors, config.json and colorizer.onnx remain v3.0, with 3,994,676 total learned parameters. The ZeroGPU Space remains pinned to immutable v3 revision 1a9eb8af2754ad2329a24cfe50d388cb559441d0 and live inference was verified on a historical photograph. No further training job is scheduled. This is the best release among the tested candidates, not evidence that no future sub-4M model could improve.