search

wkcn / MobulaOP

A Simple & Flexible Cross Framework Operators Toolkit
MIT License
164 stars 21 forks source link

Question on ATTEN SAMPLER based on MobulaOP #60

Closed AlexHu123 closed 4 years ago

AlexHu123 commented 4 years ago

Hi, will you release the PyTorch version? And will it support multi-gpu training?

wkcn commented 4 years ago

Sorry that I will not release the PyTorch version, since I'm busy now.

wkcn commented 4 years ago

The PyTorch version has been released. https://github.com/wkcn/AttentionSampler

wkcn commented 4 years ago

Close it because the same issue in https://github.com/wkcn/AttentionSampler/issues/2