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,
|