Name | Modified | Size | Downloads / 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