位于火把顶上的高级图书馆。

neuralnet-pytorch的Python项目详细描述


简介

Python - VersionPyPI - VersionPyPI - WheelGithub - TagLicense

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

Matplotlib

Visdom

Tensorboard

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

欢迎加入QQ群-->: 979659372 Python中文网_新手群

推荐PyPI第三方库


热门话题
java Android应用程序在后台   如何使用java减去日期。sql。时间戳   在Eclipse中使用Tomcat和Maven的java抛出错误   java Runnable jar什么都不做,资源不导出   并发Java内存模型:volatiles和读/写重新排序   使用Morphia在MongoDB中进行java批量升级。可能吗?   java将自己创建的sqlite数据库复制到安卓失败   在java中使用嵌套循环解决消除?   地图上多个标记的java实时动画   java如何在J2EE web应用程序中从服务层调用DAO方法   java我只想浏览文件,并将其作为电子邮件中的文件。我应如何附加:   猜猜我在think Java中的数字是57   可以在java流中的groupingBy中使用foreach吗?   我的Java程序编译时没有任何错误,但不会完全运行