one-yolov5 icon indicating copy to clipboard operation
one-yolov5 copied to clipboard

[TODO] 开发memory_reserved算子 和 性能优化

Open ccssu opened this issue 1 year ago • 2 comments

  • [ ] - 利用profile 工具优化性能(目标: GLM仓库性能优化 https://github.com/Oneflow-Inc/libai/tree/main/projects/GLM , one-yolov5分类模型性能优化)
  • [ ] - 学习oneflow 内存管理

profile工具上手

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

ccssu avatar Mar 27 '23 02:03 ccssu