Update README.md
Browse files
README.md
CHANGED
|
@@ -25,7 +25,7 @@ The following snippet illustrates code usage:
|
|
| 25 |
!pip install mutual-implication-score
|
| 26 |
|
| 27 |
from mutual_implication_score import MIS
|
| 28 |
-
mis = MIS(device='cpu')
|
| 29 |
source_texts = ['I want to leave this room',
|
| 30 |
'Hello world, my name is Nick']
|
| 31 |
paraphrases = ['I want to go out of this room',
|
|
|
|
| 25 |
!pip install mutual-implication-score
|
| 26 |
|
| 27 |
from mutual_implication_score import MIS
|
| 28 |
+
mis = MIS(device='cpu')#cuda:0 for using cuda with certain index
|
| 29 |
source_texts = ['I want to leave this room',
|
| 30 |
'Hello world, my name is Nick']
|
| 31 |
paraphrases = ['I want to go out of this room',
|