| { | |
| "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 | |
| } | |