文件名称:runx:深度学习实验管理
文件大小:39KB
文件格式:ZIP
更新时间:2024-04-09 05:24:20
Python
runx-实验管理工具 runx有助于在进行研究时自动执行常见任务: 超参数扫描 日志记录,张量板,检查点管理 实验总结 代码检查点 唯一的,每次运行的目录创建 目录 快速入门安装 用pip安装: > pip install runx 使用源安装: > git clone https://github.com/NVIDIA/runx > cd runx > python setup.py . 简介范例 假设您有一个现有项目,您可以按以下方式调用该项目: > python train.py --lr 0.01 --solver sgd 要运行超参数扫描,通常必须编写一个一次性脚本来生成扫描。但是使用runx,您只需定义一个yaml,即可定义要使用的超参数列表。 首先创建一个名为sweep.yml的yaml文件: CMD : ' python train.py ' HPARAMS :
【文件预览】:
runx-master
----requirements_dev.txt(40B)
----LICENSE(1KB)
----requirements.txt(94B)
----examples()
--------mnist_multi.yml(216B)
--------imgnet.yml(71B)
--------mnist.py(6KB)
--------imgnet.py(16KB)
--------mnist.yml(88B)
--------README.md(370B)
--------.runx(664B)
----runx()
--------utils.py(8KB)
--------test()
--------sumx.py(12KB)
--------__init__.py(0B)
--------logx.py(12KB)
--------farm.py(5KB)
--------runx.py(13KB)
--------config.py(201B)
--------collections.py(2KB)
----setup.py(836B)
----README.md(17KB)