Fast CUDA implementation ofsoft-DTWfor PyTorch. Based onpytorch-softdtwbut can run up to 100x faster! Bothforward()andbackward()passes are implemented using CUDA. My implementation is partly inspired by"Developing a pattern discovery method in time series data and its GPU acceleration"wherein a ...
您还可以运行随附的事件探查器/测试(已通过Python v3.6测试),并查看获得的加速效果: git clone https://github.com/Maghoumi/pytorch-softdtw-cuda cd pytorch-softdtw-cuda python soft_dtw_cuda.py 用法示例 脚本中已经提供了示例代码。 这是一个简单的例子: from soft_dtw_cuda import SoftDTW # Crea ...