gensen icon indicating copy to clipboard operation
gensen copied to clipboard
verified publisher icon Maluuba

fixed Encoder(When using vocab_expansion) and GensenSingle for CPU

Open najafmurtaza opened this issue 5 years ago • 1 comments

  1. GensenSingle was loading model directly to GPU instead of checking with self.cuda().
  • When using vocab_expansion
  1. Encoder had bug with try,except where it was moving all tensors to GPU by default. Now added cuda attribute, which moves the data using self.cuda() checks.

najafmurtaza avatar Aug 18 '20 23:08 najafmurtaza

CLA assistant check
All CLA requirements met.

msftgits avatar Aug 18 '20 23:08 msftgits