1 Star 0 Fork 0

ZhixuanChang/HandWrittingDigitsRecognize

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
Net.py 450 Bytes
一键复制 编辑 原始数据 按行查看 历史
ZhixuanChang 提交于 2023-10-31 20:13 . 新增训练和测试代码;
import torch
from torch import nn
from torch.nn import functional as func
class Net(nn.Module):
def __init__(self):
super(Net, self).__init__()
# xw + b
self.fc1 = nn.Linear(28 * 28, 256)
self.fc2 = nn.Linear(256, 64)
self.fc3 = nn.Linear(64, 10)
return
def forward(self, x):
x = func.relu(self.fc1(x))
x = func.relu(self.fc2(x))
x = self.fc3(x)
return x
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/zhixuanchang/hand-writting-digits-recognize.git
git@gitee.com:zhixuanchang/hand-writting-digits-recognize.git
zhixuanchang
hand-writting-digits-recognize
HandWrittingDigitsRecognize
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385