Download Latest Version GPU inference in Torchscript model_ post layer norm.zip (352.2 kB)
Email in envelope

Get an email when there's a new version of SRU

Home / v2.6.0
Name Modified Size InfoDownloads / Week
Parent folder
GPU inference in Torchscript model_ post layer norm.tar.gz 2021-05-18 324.1 kB
GPU inference in Torchscript model_ post layer norm.zip 2021-05-18 352.2 kB
README.md 2021-05-18 226 Bytes
Totals: 3 Items   676.5 kB 1
  • Support GPU/CUDA inference in Torchscript model
  • Support post layer norm
  • Support custom init value for weight_c
  • Add unit tests for GPU inference
  • Add unit tests for backward()
  • Add more unit tests for Torchscript
Source: README.md, updated 2021-05-18