File size: 426 Bytes
2ee4cd6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
{
  "_class_name": "AutoencoderVQCosmos3D",
  "_quantizer_name": "FSQuantizer",
  "in_channels": 3,
  "latent_channels": 256,
  "layers_per_block": 2,
  "norm_num_groups": 1,
  "out_channels": 3,
  "sample_size": 1024,
  "sample_frames": 49,
  "num_vq_embeddings": 64000,
  "vq_embed_dim": 6,
  "patch_size": 2,
  "temporal_stride": 4,
  "spatial_stride": 8,
  "block_out_channels": [
    128,
    256,
    512,
    512
  ]
}