add support to tensorrt - Githubissues

calico / baskerville

Machine learning methods for DNA sequence analysis.

Apache License 2.0

32 stars 9 forks source link

add support to tensorrt #13

Closed lruizcalico closed 9 months ago

lruizcalico commented 10 months ago

Description of your changes

Add support for tensorrt

Issue ticket number and link

Type of change

[ ] Bug fix
[x] New feature
- [x] Backwards Incompatible?
[ ] Refactoring / code clean-up
[ ] Documentation add / update
[ ] Automated Test
[ ] Other (please specify)

(If applicable) How has this been tested?

Tested on VM with T4 GPU. Input files can be from both local and gcs