Update README.md
Browse files
README.md
CHANGED
|
@@ -23,6 +23,13 @@ PreceptCLIP-Memorability is a model designed to predict image memorability (the
|
|
| 23 |
- *Learning Rate*: 5e-05
|
| 24 |
- *Batch Size*: 32
|
| 25 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
## Usage
|
| 27 |
|
| 28 |
To use the model for inference:
|
|
|
|
| 23 |
- *Learning Rate*: 5e-05
|
| 24 |
- *Batch Size*: 32
|
| 25 |
|
| 26 |
+
## Requirements
|
| 27 |
+
- python=3.9.15
|
| 28 |
+
- cudatoolkit=11.7
|
| 29 |
+
- torchvision=0.14.0
|
| 30 |
+
- transformers=4.45.2
|
| 31 |
+
- peft=0.14.0
|
| 32 |
+
|
| 33 |
## Usage
|
| 34 |
|
| 35 |
To use the model for inference:
|