下拉刷新
项目详情
navbar_avatar分享
repo_avatar
HelloGitHub 评分
0 人评分
一款小型的开源深度学习框架
开源MIT
认领
收藏
分享
26.7k
星数
中文
Python
主语言
活跃
355
贡献者
212
Issues
组织
0.9.2
最新版本
3k
Forks
MIT
协议
更多
tinygrad image
一款小型的开源深度学习框架。它代码不足 1k 行足够简单,支持深度模型推理与训练。示例代码: ```python from tinygrad.tensor import Tensor import tinygrad.optim as optim class TinyBobNet: def __init__(self): self.l1 = Tensor.uniform(784, 128) self.l2 = Tensor.uniform(128, 10) def forward(self, x): return x.dot(self.l1).relu().dot(self.l2).logsoftmax() model = TinyBobNet() optim = optim.SGD([model.l1, model.l2], lr=0.001) # ... and complete like pytorch, with (x,y) data out = model.forward(x) loss = out.mul(y).mean() optim.zero_grad() loss.backward() optim.step() ```
收录于:
第 60 期
标签:
深度学习
AI
Python

评论

评分:
暂无精选评论