forked from VainF/Torch-Pruning
-
Notifications
You must be signed in to change notification settings - Fork 1
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
oneflow适配调研 #1
Comments
AttributeError: module 'oneflow.nn.modules.conv' has no attribute '_ConvNd'项目使用处
Renferce |
AttributeError: module 'oneflow.nn' has no attribute 'MultiheadAttention'项目使用处nn.MultiheadAttention: multihead_attention_counter_hook Renferce |
感觉我们可以改成TORCH_CONV = nn.modules.Conv1D or nn.modules.Conv2D or nn.modules.Conv1D 这样子? |
Oneflow-Inc/oneflow#10086 short和char这里支持了 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
本报告调研了github地址为https://github.com/VainF/Torch-Pruning 的仓库。
主要集中在分析使用oneflow重构仓库中的代码所需的工作量和难点,以及可能遇到的问题
仓库简介
主要目的: is a versatile library that enables structural network pruning for a wide range of neural networks, including Vision Transformers, Yolov7, FasterRCNN, SSD, ResNet, DenseNet, ConvNext, RegNet, ResNext, FCN, DeepLab, VGG, etc.
项目目录介绍:
benchmarks
[源仓库在计划开发中]https://github.com/Oneflow-Inc/OneFlow-Pruning#plans
tests 测试
torch_pruning 定义的剪枝工具
tutorials
适配开发暂时集中在 test torch_pruning tutorials中。
仓库功能
https://github.com/Oneflow-Inc/OneFlow-Pruning#features
oneflow适配缺失模块
结论
在不考虑正确性修复上面模块,如链接所示 Oneflow-Inc/oneflow@master...adapt_to_torch_pruning
测试了 Torch-Pruning/tutorials/0 - QuickStart.ipynb 可以正常执行
测试时运行输出
plan
逐一解决上面发现的问题,进行仓库适配
The text was updated successfully, but these errors were encountered: