one-yolov5
one-yolov5 copied to clipboard
[TODO] 开发memory_reserved算子 和 性能优化
- [ ] - 利用profile 工具优化性能(目标: GLM仓库性能优化 https://github.com/Oneflow-Inc/libai/tree/main/projects/GLM , one-yolov5分类模型性能优化)
- [ ] - 学习oneflow 内存管理
profile工具上手
- nsys: 结合NVTX注释上手nsys
- py-spy https://github.com/Oneflow-Inc/one-yolov5/issues/120#issue-1616558768
one-yolov5项目
项目地址: https://github.com/Oneflow-Inc/one-yolov5 数据集路径: @oneflow-25:/data/home/fengwen/imagenette160 权重路径: @oneflow-25:/data/home/fengwen/weight_v1_2_0
如果执行nsys产生报错
The target application terminated. One or more process it created re-parented.
Waiting for termination of re-parented processes.
Use the `--wait` option to modify this behavior.
请将 train.py中 check_git_status() 这一行注释
glm 项目
项目地址: https://github.com/Oneflow-Inc/libai/tree/main/projects/GLM 权重路径: @oneflow-25:/data/home/xiezipeng/glm-10b-chinese @oneflow-25:/data/home/xiezipeng/glm-10b