位于火把顶上的高级图书馆。
neuralnet-pytorch的Python项目详细描述
简介
pytorch通用神经网络的高级框架。
就个人而言,从西亚诺到火把城就像从90年代到现代的时间旅行。 然而,尽管有很多钟声和口哨声,我仍然觉得pytorch缺少了一些元素。 被证实从未加入图书馆。 因此,编写这个库是为了给当前的魔法火把添加更多功能。这里的所有模块 直接从pytorch中派生出相应的模块,所以一切都应该很熟悉。例如, 下面是pytorch中的片段
from torch import nn
model = nn.Sequential(
nn.Conv2d(1, 20, 5, padding=2),
nn.ReLU(),
nn.Conv2d(20, 64, 5, padding=2),
nn.ReLU()
)
可以在neuralnet pytorch中重写为
import neuralnet_pytorch as nnt
model = nnt.Sequential(
nnt.Conv2d(1, 20, 5, padding='half', activation='relu'),
nnt.Conv2d(20, 64, 5, padding='half', activation='relu')
)
与本地pytorch相同,或
import neuralnet_pytorch as nnt
model = nnt.Sequential(input_shape=1)
model.add_module('conv1', nnt.Conv2d(model.output_shape, 20, 5, padding='half', activation='relu'))
model.add_module('conv2', nnt.Conv2d(model.output_shape, 64, 5, padding='half', activation='relu'))
当你在另一个图层上添加一个图层时,它可以让你从大量的记忆和手工计算中解脱出来。 西亚诺人也会发现一些回忆,因为许多功能是高度启发西亚诺。
要求
Pytorch>;=1.0.0
Gin-config(可选)
安装
稳定版本
pip install --upgrade neuralnet-pytorch
出血边缘型
pip install git+git://github.com/justanhduc/neuralnet-pytorch.git@master
< P>使用一些已收集的CUDA/C+OPS安装版本,使用pip install git+git://github.com/justanhduc/neuralnet-pytorch.git@fancy
用法
手册参考仍在开发中,可在https://neuralnet-pytorch.readthedocs.io找到。
待办事项
- [X]添加简介和安装
- [X]编写文档
- []添加示例
免责声明
这个包是我在读博士期间的一点空闲时间的产物, 因此,大多数(但不是所有)编写的模块都得到了正确的检查。 无需更换或退款。 欢迎所有公关。
致谢
cuda倒角距离取自AtlasNetrepo。
adabound优化器取自其official repo。
为pytorch代码改编的gin取自Gin-config。
监控方案的灵感来自WGAN。