Instructions to use Lutech-AI/I-SPIn with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Lutech-AI/I-SPIn with Transformers:
# Load model directly from transformers import ISPIn model = ISPIn.from_pretrained("Lutech-AI/I-SPIn", dtype="auto") - Notebooks
- Google Colab
- Kaggle
Commit ·
af83f90
1
Parent(s): 2cad7e4
Update README.md
Browse files
README.md
CHANGED
|
@@ -56,7 +56,7 @@ print(predictions) # -> [0 -1]
|
|
| 56 |
The computation is subdivided in two tasks (embedding, classification) to simplify a custom fine-tuning process.
|
| 57 |
|
| 58 |
If you want to further optimize this classification head, you might want to deepcopy the layers and continue training
|
| 59 |
-
(
|
| 60 |
```python
|
| 61 |
import torch
|
| 62 |
import copy
|
|
@@ -91,8 +91,6 @@ ISPIn(
|
|
| 91 |
)
|
| 92 |
```
|
| 93 |
|
| 94 |
-
If you want to take
|
| 95 |
-
|
| 96 |
## Evaluation results
|
| 97 |
| Dataset | Metric | Performance |
|
| 98 |
|:--------------------------------------:|--------------|-------------|
|
|
|
|
| 56 |
The computation is subdivided in two tasks (embedding, classification) to simplify a custom fine-tuning process.
|
| 57 |
|
| 58 |
If you want to further optimize this classification head, you might want to deepcopy the layers and continue training
|
| 59 |
+
(one can choose which layers by slicing the list):
|
| 60 |
```python
|
| 61 |
import torch
|
| 62 |
import copy
|
|
|
|
| 91 |
)
|
| 92 |
```
|
| 93 |
|
|
|
|
|
|
|
| 94 |
## Evaluation results
|
| 95 |
| Dataset | Metric | Performance |
|
| 96 |
|:--------------------------------------:|--------------|-------------|
|