NiGuLa commited on
Commit
dd1a0b3
1 Parent(s): f55df64

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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',