PHP前端开发

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

百变鹏仔 3天前 #Python
文章标签 我的机器

为我的机器学习库添加了调试模式、调试绘图参数、调试层梯度、部署、修复库的子模块,该库应该是 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 okrolearnokrolearn.Tensor() # Example usage