首页 > 文章列表 > 为我的机器学习库(如 pytorch)添加了高级调试功能

为我的机器学习库(如 pytorch)添加了高级调试功能

231 2024-08-13

为我的机器学习库(如 pytorch)添加了高级调试功能

为我的机器学习库添加了调试模式、调试绘图参数、调试层梯度、部署、修复库的子模块,该库应该是 scikit learn 和 pytorch 的原始组合。

https://pypi.org/project/okrolearn/

安装

要为 cpu 安装 okrolearn,请使用: pip install okrolearn==0.2.5
要安装 gpu,请执行 pip install okrolearn 这需要 cuda

要在代码中使用 okrolearn,请导入主模块:

from okrolearn import okrolearn
okrolearn.Tensor() # Example usage
来源:https://dev.to/okerew/added-advanced-debugging-features-to-my-machine-learning-library-like-pytorch-2m37

本类最新

查看更多