Skip to main content

深度学习框架(Deep Learning Framework)

Project description

深度学习框架 V0.1.2(pip install start-zero)

一、未来规划

1、修正BUG、优化代码、减少内存使用率
2、优化、完善Define-by-Run体系
3、优化、完善CPU和GPU的转化机制
4、提供训练参数的保存和加载
5、增加函数、层、模型、优化器
6、增加对CNN(卷积神经网络)、RNN(循环神经网络)等的实现

二、框架主要内容

序号 内容 备注
1 数值微分、自动微分、高阶求导 高阶求导使用反向传播的反向传播
2 处理标量和张量 Tensor类(兼容numpy)
3 支持GPU运算 只支持N卡,目前代码需要将np.array变为cupy.array且不能灵活切换(待完善转换机制)
使用条件:
①Config.ENABLE_GPU = True
②CUDA.is_available() == True
4 Define-by-Run 运行时动态图(核心),静态图(Define-and-Run)暂时不会去实现,因为涉及领域特定语言
5 函数 目前一共支持41个函数(34个函数有正向传播和反向传播)
6 目前一共支持1个层
7 模型 目前一共支持1个模型
8 优化器 目前一共支持5个优化器

三、支持的函数

序号 函数 符号 备注
1 合并求和、广播、求和、平均数、矩阵相乘、矩阵转置、重塑形状、线性回归、切片、切片(梯度) sum_to、broadcast_to、sum、average、matmul、transpose、reshape、linear、get_item、get_item_grad ft0.py(10个函数)
2 加、减、乘、除、幂、负数、模 add(+)、sub(-)、mul(*)、div(/)、power(**)、neg(-)、mod(%) ft1.py(7个函数)
3 正弦、余弦、正切、双曲正切 sin、cos、tan、tanh ft2.py(4个函数)
4 e为底的指数、10为底的对数、e为底的对数 exp、lg、ln ft3.py(3个函数)
5 S型生长曲线、线性整流函数、归一化指数函数、归一化指数函数、线性整流函数、阶跃函数 sigmoid、relu、softmax、log_softmax、leaky_relu、step ft4.py(6个函数)
6 均方误差、交叉熵损失、交叉熵损失、二元交叉熵 mean_squared_error、softmax_cross_entropy、sigmoid_cross_entropy、binary_cross_entropy ft5.py(4个函数)
7 最大值、最小值、限定数组上下界、批量 max、min、clip、batch_norm ft6.py(4个函数)
8 准确度、退出、嵌入ID accuracy、dropout、embed_id ft7.py(3个函数)

四、支持的层

序号 符号 备注
1 线性层 LinearLayer

五、支持的模型

序号 模型 符号 备注
1 多层感知器 MLP

六、支持的优化器

序号 优化器 符号 备注
1 随机梯度下降 SGD
2 动量梯度下降 MomentumSGD
3 梯度下降优化 AdaGrad
4 梯度下降优化 AdaDelta
5 梯度下降优化 Adam

七、发布到PyPI

1、登录PyPI官网并完成账号注册和安全认证
2、安装插件(如:E:\pyhton\python.exe -m pip install --upgrade pip setuptools wheel、E:\pyhton\python.exe -m pip install twine)
3、生成压缩包(python setup.py sdist)
4、上传压缩包(如:E:\pyhton\python.exe -m twine upload dist/*)
注:完成2FA认证后,username是:token,password是:生成的token
5、安装和卸载(pip install start-zero、pip uninstall start-zero)
注:指定版本如:pip install start-zero==1.0.0,也可以<或<=等

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

start-zero-0.1.2.tar.gz (22.6 kB view details)

Uploaded Source

File details

Details for the file start-zero-0.1.2.tar.gz.

File metadata

  • Download URL: start-zero-0.1.2.tar.gz
  • Upload date:
  • Size: 22.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/4.0.2 CPython/3.10.6

File hashes

Hashes for start-zero-0.1.2.tar.gz
Algorithm Hash digest
SHA256 d2fba767c1528474508465715405838a08cae19e8236b92f0ce02b630e56fbdc
MD5 4430d9154a626620bf623f11ea08e8b1
BLAKE2b-256 85b91d6b1945050b20571a5e95ce7f615a62347422dda452e1ad1faa8723b93b

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page