文件名称:nas-framework:神经架构搜索框架
文件大小:81KB
文件格式:ZIP
更新时间:2024-06-04 11:08:11
Python
神经架构搜索框架 顾名思义,NAS框架是一个有助于在各种数据集上进行的框架。 它提供了一种简单灵活的方法来定义任意复杂度的搜索空间和Architect类,该类无需修改在模板之后定义的任何搜索空间即可工作。 架构师是一种递归神经网络,它通过递归地创建目前预测的计算图的表示并根据接收到该表示的策略选择一个动作(即搜索空间的特定维度中的点)来生成计算图的描述。输入。 架构师接受了使用强化学习的培训,特别是在各种数据集上的类似于Actor-Critic的算法变体。 该软件包当前处于alpha版本,并且开箱即用地支持“多层感知器”和“递归神经网络”搜索空间。 如果这两个搜索空间可以满足您的需求,那么为了执行神经体系结构搜索,您要做的torch.utils.data.Dataset使用数据创建torch.utils.data.Dataset并在scripts.train_toxic修改toxi
【文件预览】:
nas-framework-master
----docs()
--------Makefile(635B)
--------source()
----tests()
--------mlp_space.py(9KB)
--------coach_base.py(818B)
--------test_coaches.py(1KB)
--------__init__.py(0B)
--------test_storage.py(5KB)
--------rnn_space.py(5KB)
--------architect.py(2KB)
----LICENSE(1KB)
----config.yml(934B)
----nasframe()
--------storage.py(12KB)
--------searchspaces()
--------loaders()
--------__init__.py(107B)
--------coaches()
--------architect.py(17KB)
--------utils()
--------scripts()
----requirements()
--------basic.txt(199B)
--------all.txt(267B)
--------docs.txt(23B)
----setup.py(776B)
----.gitignore(2KB)
----README.md(3KB)