rookierufus commited on
Commit
d8238a4
·
verified ·
1 Parent(s): dd01134

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +28 -0
README.md ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Compresser Encoder (Perceiver Resampler)
2
+
3
+ Phase 0 pretrained Perceiver for the Mamba-3 Semantic Video Compressor.
4
+
5
+ ## Architecture
6
+ - **Type**: Perceiver Resampler (cross-attention compressor)
7
+ - **Input**: [B, 576, 1664] — V-JEPA 2.1 ViT-Gigantic patch latents
8
+ - **Output**: [B, 64, 512] — compressed tokens
9
+ - **Params**: ~20.6M
10
+ - **Details**: 64 learned queries, 6 cross-attention layers, 16 heads, FFN 512→2048
11
+
12
+ ## Training
13
+ - **Dataset**: [Vjepa_mamba_dataset_v2](https://huggingface.co/datasets/rookierufus/Vjepa_mamba_dataset_v2) (50 hours video, 384×384, 8fps)
14
+ - **V-JEPA**: Frozen [vjepa2_1_vit_gigantic_384](https://github.com/facebookresearch/vjepa2) (2.2B params)
15
+ - **Loss**: MSE reconstruction via autoencoder (Perceiver → Decoder → V-JEPA latent)
16
+ - **Optimizer**: AdamW, lr=1e-4, cosine to 1e-6
17
+ - **Hardware**: RTX 4090 (48 GB), bf16
18
+
19
+ ## Usage
20
+ ```python
21
+ from model.models.perceiver import PerceiverResampler
22
+
23
+ model = PerceiverResampler(input_dim=1664, output_dim=512, num_queries=64)
24
+ model.load_state_dict(torch.load("perceiver_stepX_hrsY.pt"))
25
+ # Input: [B, 576, 1664] V-JEPA latents → Output: [B, 64, 512]
26
+ ```
27
+
28
+ Part of the Mamba-3 Semantic Video Compressor pipeline.