仅用Python和Numpy完成深度学习框架(附代码实现)

使用Numpy完成深度学习框架

本文针对自己使用Numpy搭建的深度学习框架进行了介绍。
代码地址在:使用Numpy实现的简单深度学习框架

框架结构

在深度学习中,有几个基本的重要组件:1. 基础的网络模块,如全连接层,卷积层,激活函数,BN层等,它们组建起了网络的基础结构。2. 优化器,如SGD,Adam,RMSprop等,它们负责将网络中各个层的参数进行更新。3. Loss,这决定了网络所要优化的目标,并且负责在每次计算Loss后进行反向传播来计算各个层的梯度。

因此,我将这几类作为了框架的基础父类:
在Model中,我们定义了该层的参数,梯度和是否为Validation。也包含了前向传播方法,后向传播方法,和参数初始化方法。

class Model():
    def __init__(self,valid=False):
        self.valid=valid
        self.params=OrderedDict()
        self.gradient=OrderedDict()
    def forward(self):
        pass
    def backward(self):
        pass
    def val(self):
        self.valid=True
    def train(self):
        self.valid=False
    def __intweight(self):
        pass

在Optimizer中,需要在初始化时传入模型,以此来得到模型中的各个参数。step用于根据网络中各个层计算好的梯度,对该层的权重进行更新。

class Optimizer():
    def __init__(self,model):
        self.model=model
    def step(self):
        pass
    def zero_grad(self):
        pass

在Loss中,需要传入模型,batch size,网络预测值和标签。用于计算损失和梯度。

class Loss():
    def __init__(self,model,batch_size):
        self.model=model
        self.y=None
        self.label=None
        self.batch_size=batch_size
    def loss(self,y,label):
        pass
    def backward
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值