项目详情
由分享
HelloGitHub 评分
0 人评分
开源•MIT
认领
讨论
收藏
分享
26.7k
星数
否
中文
Python
主语言
是
活跃
355
贡献者
212
Issues
是
组织
0.9.2
最新版本
3k
Forks
MIT
协议
更多
一款小型的开源深度学习框架。它代码不足 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()
```
评论
评分:
暂无精选评论