Pan Wang
commited on
Commit
·
aa46ef0
1
Parent(s):
59ebedd
Update README.md
Browse files
README.md
CHANGED
|
@@ -22,15 +22,19 @@ The framework of DLF. Please refer to [Paper Link](arxiv) for details.
|
|
| 22 |
- CUDA 11.7
|
| 23 |
|
| 24 |
### Installation
|
| 25 |
-
- Create conda environment. Please make sure you have installed conda before.
|
| 26 |
```
|
| 27 |
conda create -n DLF python==3.9.13
|
| 28 |
```
|
| 29 |
```
|
| 30 |
pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117
|
| 31 |
```
|
|
|
|
| 32 |
```
|
| 33 |
-
|
|
|
|
|
|
|
|
|
|
| 34 |
```
|
| 35 |
|
| 36 |
### Datasets
|
|
@@ -41,7 +45,7 @@ Please note that the meta information and the raw data are not available due to
|
|
| 41 |
### Run the Codes
|
| 42 |
- Training
|
| 43 |
|
| 44 |
-
You can first set the
|
| 45 |
```
|
| 46 |
python3 train.py
|
| 47 |
```
|
|
@@ -53,11 +57,11 @@ You can first set the testing dataset name in `./test.py` as "mosei" or "mosi",
|
|
| 53 |
```
|
| 54 |
python3 test.py
|
| 55 |
```
|
| 56 |
-
We also provide
|
| 57 |
|
| 58 |
|
| 59 |
### Citation
|
| 60 |
-
If you find the code and our idea helpful in your
|
| 61 |
|
| 62 |
```
|
| 63 |
@article{wang2025dlf,
|
|
|
|
| 22 |
- CUDA 11.7
|
| 23 |
|
| 24 |
### Installation
|
| 25 |
+
- Create a conda environment. Please make sure you have installed conda before.
|
| 26 |
```
|
| 27 |
conda create -n DLF python==3.9.13
|
| 28 |
```
|
| 29 |
```
|
| 30 |
pip install torch==1.13.0+cu117 torchvision==0.14.0+cu117 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu117
|
| 31 |
```
|
| 32 |
+
- Clone this repo.
|
| 33 |
```
|
| 34 |
+
git clone https://github.com/pwang322/DLF.git
|
| 35 |
+
```
|
| 36 |
+
```
|
| 37 |
+
pip install requirements.txt
|
| 38 |
```
|
| 39 |
|
| 40 |
### Datasets
|
|
|
|
| 45 |
### Run the Codes
|
| 46 |
- Training
|
| 47 |
|
| 48 |
+
You can first set the training dataset name in `./train.py` as "mosei" or "mosi", and then run:
|
| 49 |
```
|
| 50 |
python3 train.py
|
| 51 |
```
|
|
|
|
| 57 |
```
|
| 58 |
python3 test.py
|
| 59 |
```
|
| 60 |
+
We also provide pre-trained models for testing. ([Google drive](https://drive.google.com/drive/folders/1GgCfC1ITAnRRw6RScGc7c2YUg5Ccbdba?usp=sharing))
|
| 61 |
|
| 62 |
|
| 63 |
### Citation
|
| 64 |
+
If you find the code and our idea helpful in your research or work, please cite the following paper.
|
| 65 |
|
| 66 |
```
|
| 67 |
@article{wang2025dlf,
|