search

chenjianqu / RAFT_CPP

The optical flow algorithm RAFT implemented with C++(Libtorch+TensorRT)
MIT License
40 stars 4 forks source link

多个engine #5

Open Cai-RS opened 8 months ago

Cai-RS commented 8 months ago

你好，感谢你的工作！ RAFT是由多个部分组成，我好奇这里的feature encoder网络和context encoder网络得到的engine能否在GPU上同时运行呢？如果可以的话，应该可以覆盖其中一个的时间？这两个卷积网络似乎都是比较轻量级的，如果是2080Ti也许可以异步起来？