search

Dao-AILab / causal-conv1d

Causal depthwise conv1d in CUDA, with a PyTorch interface
BSD 3-Clause "New" or "Revised" License
335 stars 61 forks source link

Export causal-conv1d models #40

Open bhack opened 1 week ago

bhack commented 1 week ago

Can you add a small example on how to export causal-conv1d models?

See: https://pytorch.org/tutorials/intermediate/torch_export_tutorial.html#custom-ops