training/run_config.json

{
  "init_from": "init",
  "revision": null,
  "parquet": "/tmp/colorizer/combined.parquet",
  "manifest": "manifest.json",
  "output": "run",
  "steps": 1122,
  "size": 256,
  "batch_size": 32,
  "lr": 1e-05,
  "context_lr": 1e-05,
  "head_lr": 1e-05,
  "rebalance_lambda": 0.7,
  "consistency": 0.0,
  "statistics": "stats.npz",
  "freeze_bn": true,
  "seed": 123,
  "threads": 4,
  "workers": 4,
  "eval_every": 374,
  "push_to_hub": false,
  "hub_model_id": "User-2468/mini-unet-colorizer",
  "device": "cuda",
  "amp_dtype": "torch.bfloat16",
  "parameters": 3968892,
  "bin_sha256": "4bcdeb9ad2dcf6d460e7157e436a524b6b885ef8b5552a2f0480550db811e173"
}