重写PyTorch框架,帮助您学习AI/ML
edutorch的Python项目详细描述
教育火炬
重写Pythorch框架,旨在帮助您学习AI/ML!在
Pythorch是构建和训练深层神经网络的最惊人的框架之一。它最大的优势之一是为构建和训练这些模型提供了一个直观和可扩展的界面。在
在这个项目中,我提供了我自己版本的PyTorch框架,旨在帮助您理解关键概念。目标是提供流行层、模型和优化器的显式实现。最重要的是,这段代码被设计成可读性和清晰性。这些例子中有许多是从斯坦福大学的cs230/231N课程材料中修改而来的。在
爱托奇vs Pythorch
Eduthorch和Pythorch的一个显著区别是Eduthorch不提供autograd。你自己推导/实现backprop步骤有很多教育上的好处,如果你想自动计算梯度,你最好使用真正的框架。另外,如果您想了解autograd系统是如何实现的,可以查看andrejkarpathy的micrograd project。在
出于同样的原因,也没有CUDA或GPU支持。在
贡献
所有问题和拉请求是非常感谢!在
- 首先,确保运行
scripts/install-hooks
。在 - 要运行所有测试并使用自动格式化工具,请查看
scripts/run-tests
。在 - 要只运行单元测试,请运行
pytest
。在
- 项目
标签: