Skip to content

Upgrade to PyTorch 0.3.0 and minor changes

Compare
Choose a tag to compare
@kmkurn kmkurn released this 29 Dec 13:37
· 123 commits to master since this release

Upgrade to PyTorch 0.3.0

PyTorch 0.3.0 is out! We upgrade our PyTorch dependency to version 0.3.0, so now we can use its fancy indexing to get transition score instead of manually broadcasting tensors.

Others

  • Fix summing mask to get length by first converting it to LongTensor to avoid overflow.
  • Specify dim explicitly when squeeze-ing to prevent squeezing unintended dimensions.