快速入门

快速入门主要介绍如何使用Mytorch搭建神经网络和进行训练

模块导入

from Mytorch.trainer.trainer import Trainer
from Mytorch.data.data_loader import MnistData
from Mytorch.torch.mlp import MLP

加载数据和模型

在这里我们导入了small_mnist数据和MLP模型。
更多数据在
更多模型可以在

data = MnistDat("\dataset\mnist_small_matlabmat")
model =MLP()
model.show()

mlp

导入trainer 进行训练

trainer =Trainer(max_epochs=10)
trainer.fit(model,data)

Epoch: 10/10, loss: 0.009  train_acc: 0.999  test_acc: 0.969  examples/sec:94.9

output