Function write_snapshot_file

training/mlp_train.py:195–229

(
    snapshots: Sequence[LayerSnapshot],
    directory: Path,
    order: int,
    identifier: str,
)

def write_snapshot_file(
    snapshots: Sequence[LayerSnapshot],
    directory: Path,
    order: int,
    identifier: str,
) -> Path:
    slug = slugify_identifier(identifier)
    filename = f"{order:03d}_{slug}.json"
    path = directory / filename
    layers_payload = []
    for snapshot in snapshots:
        meta = snapshot.metadata
        layers_payload.append(
            {
                "layer_index": meta.layer_index,
                "name": meta.name,
                "activation": meta.activation,
                "weights": {
                    "shape": list(meta.weight_shape),
                    "data": tensor_to_base64(snapshot.weight),
                },
                "biases": {
                    "shape": list(meta.bias_shape),
                    "data": tensor_to_base64(snapshot.bias),
                },
            }
        )
    payload = {
        "version": 1,
        "dtype": "float16",
        "layers": layers_payload,
    }
    directory.mkdir(parents=True, exist_ok=True)
    path.write_text(json.dumps(payload, separators=(",", ":")))
    return path


def to_posix_relative(path: Path, base: Path) -> str: