ae_v31_d16c32.encoder.json

{
  "latent_c": 32,
  "latent_down": 16,
  "terrain_cond": true,
  "upsample_decoder": true,
  "source": "/workspace/runs/ae_v31_d16c32/ae_v3.pt",
  "format": "spatial_ae_encoder_v3",
  "keys": [
    "enc_fuse.0.0.bias",
    "enc_fuse.0.0.weight",
    "enc_fuse.0.1.bias",
    "enc_fuse.0.1.weight",
    "enc_fuse.1.bias",
    "enc_fuse.1.weight",
    "enc_stem.0.0.bias",
    "enc_stem.0.0.weight",
    "enc_stem.0.1.bias",
    "enc_stem.0.1.weight",
    "enc_stem.1.0.bias",
    "enc_stem.1.0.weight",
    "enc_stem.1.1.bias",
    "enc_stem.1.1.weight",
    "enc_stem.2.0.bias",
    "enc_stem.2.0.weight",
    "enc_stem.2.1.bias",
    "enc_stem.2.1.weight",
    "enc_stem.3.0.bias",
    "enc_stem.3.0.weight",
    "enc_stem.3.1.bias",
    "enc_stem.3.1.weight",
    "enc_stem.4.0.bias",
    "enc_stem.4.0.weight",
    "enc_stem.4.1.bias",
    "enc_stem.4.1.weight",
    "owner_emb.weight"
  ],
  "num_tensors": 27
}