easytorch
使用Python的numpy实现的简易深度学习框架,API与pytorch基本相同,实现了自动求导、基础优化器、layer等。
1 文档目录
2 Quick Start
from easytorch.layer import Linear, Tanh, Sequential
from easytorch.optim import SGD
import easytorch.functional as F
# Create a model, optimizer, loss function
model = Sequential(
Linear(1, 5),
Tanh(),
Linear(5, 1)
)
opt = SGD(model.parameters(), lr=3e-4)
loss_fn = F.mse_loss
# train the mod
1