Update README.md
Browse files
README.md
CHANGED
@@ -3,8 +3,11 @@ license: apache-2.0
|
|
3 |
---
|
4 |
# EgoSpeak Checkpoints
|
5 |
|
6 |
-
This repo
|
|
|
|
|
|
|
|
|
|
|
|
|
7 |
|
8 |
-
- **Datasets**: EasyCom, Ego4D, Pretrain, Pretrain+EasyCom, Pretrain+Ego4D
|
9 |
-
- **Modalities**: Audio (A), Video (V), Audio+Video (AV)
|
10 |
-
- **Models**: Transformer, GRU, Mamba
|
|
|
3 |
---
|
4 |
# EgoSpeak Checkpoints
|
5 |
|
6 |
+
This repo contains final checkpoints for the EgoSpeak project.
|
7 |
+
- **Models**: `lstr`, `mamba`, `miniroad`
|
8 |
+
- **Datasets**: `easycom`, `ego4dshuffle`, `ytconv_pretrained`
|
9 |
+
- **Modalities**: `A` (audio), `V` (video), `AV` (audio+video)
|
10 |
+
|
11 |
+
## File Naming Convention
|
12 |
+
- `mamba_easycom_A.pth` → Mamba model, trained on EasyCom, audio-only
|
13 |
|
|
|
|
|
|