Update README.md
Browse files
README.md
CHANGED
|
@@ -24,13 +24,13 @@ The checkpoints are organized by experiment configuration. We use the following
|
|
| 24 |
### Checkpoint Variations
|
| 25 |
| Folder Name | Method Description | Alignment Strategy |
|
| 26 |
| :--- | :--- | :--- |
|
| 27 |
-
| `192_l2_cube_all` | **
|
| 28 |
| `192_l2_cube_all_nm` | Default (No Masking) | L2 Similarity Tensor |
|
| 29 |
| `192_l2_cube_pc` | Frozen Image/Text | L2 Similarity Tensor |
|
| 30 |
| `192_cos_cube_all` | Cosine Variant | Cosine Similarity Tensor |
|
| 31 |
| `192_cos_matrix_all` | Pairwise Matrix | 3× Pairwise Similarity Matrices |
|
| 32 |
| `192_cos_matrix_pc` | Pairwise (Frozen) | 3× Pairwise Similarity Matrices |
|
| 33 |
-
| `
|
| 34 |
|
| 35 |
## How to Load the Models
|
| 36 |
|
|
@@ -48,7 +48,7 @@ from huggingface_hub import hf_hub_download
|
|
| 48 |
|
| 49 |
REPO_ID = "Ximeng0831/CTP"
|
| 50 |
# Example: Loading the default proposed model
|
| 51 |
-
SUBFOLDER = "192_l2_cube_all"
|
| 52 |
FILENAME = "ckpt_epoch9.pt"
|
| 53 |
|
| 54 |
checkpoint_path = hf_hub_download(
|
|
|
|
| 24 |
### Checkpoint Variations
|
| 25 |
| Folder Name | Method Description | Alignment Strategy |
|
| 26 |
| :--- | :--- | :--- |
|
| 27 |
+
| `192_l2_cube_all` | **Default** | L2 Similarity Tensor |
|
| 28 |
| `192_l2_cube_all_nm` | Default (No Masking) | L2 Similarity Tensor |
|
| 29 |
| `192_l2_cube_pc` | Frozen Image/Text | L2 Similarity Tensor |
|
| 30 |
| `192_cos_cube_all` | Cosine Variant | Cosine Similarity Tensor |
|
| 31 |
| `192_cos_matrix_all` | Pairwise Matrix | 3× Pairwise Similarity Matrices |
|
| 32 |
| `192_cos_matrix_pc` | Pairwise (Frozen) | 3× Pairwise Similarity Matrices |
|
| 33 |
+
| `192_cos_matrix_IP_pc`| Image-Point Only | 1× Similarity Matrix (I-L) |
|
| 34 |
|
| 35 |
## How to Load the Models
|
| 36 |
|
|
|
|
| 48 |
|
| 49 |
REPO_ID = "Ximeng0831/CTP"
|
| 50 |
# Example: Loading the default proposed model
|
| 51 |
+
SUBFOLDER = "pointnet2/192_l2_cube_all"
|
| 52 |
FILENAME = "ckpt_epoch9.pt"
|
| 53 |
|
| 54 |
checkpoint_path = hf_hub_download(
|