Merge pull request #2 from heartcored98/master

Fix misspelling when checking cuda availbility
theeluwin · Jun 3, 2018 · f01b465 · f01b465
2 parents f52343b + 4bda45a
commit f01b465
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model.py b/model.py
@@ -50,7 +50,7 @@ def forward_i(self, data):
 
     def forward_o(self, data):
         v = V(LT(data), requires_grad=False)
-        v = v.cuda() if self.ivectors.weight.is_cuda else v
+        v = v.cuda() if self.ovectors.weight.is_cuda else v
         return self.ovectors(v)