Skip to content

优化process_your_own_data.py #141

Description

@1os3

process_your_own_data.py
原有版本依赖特定的自定义算子且时间复杂度是$O(N^2)$
经过优化,复杂度变为$O(N)$且速度极大提升,同时转向全GPU运算(基于Pytorch,测试中用的pytorch.xpu,只需要修改为pytorch.cuda即可兼容cuda)
值得注意的是数据类型需求发生了变化
建议合并

process_your_own_data.py

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions