用于培训深度学习交叉验证培训的工具。
trainbench的Python项目详细描述
列车工作台
Trainbench是一个工具,用于快速建立(深度学习)交叉验证培训课程,包括配置实例化、学习权重、图表等,并在每个培训阶段进行记录
与路缘石兼容
⚠65039;此包当前正在积极开发中!
开始
安装
pip install trainbench
创建列车会话
# train.py# namename='experiment_01'# parameters to cross validate againstparameters={'fc_size':[256,512,1024]# ...}deftrain(parameters):xyz=parameters['xyz']# ...
从命令行运行
trainbench .
查看结果
<name>/crosses/
/001
├─ meta.yml
├─ history.pkl
└─ weights/
├─ 001.h5
└─ 002.h5
...
在jupyter笔记本/python脚本中
fromtrainbenchimportBenchparameters={}deftrain_fn(parameters):passbench=Bench()bench.train('experiment_xyz',train_fn,parameters)
作者注释
trainbench .
查看结果
<name>/crosses/
/001
├─ meta.yml
├─ history.pkl
└─ weights/
├─ 001.h5
└─ 002.h5
...
在jupyter笔记本/python脚本中
fromtrainbenchimportBenchparameters={}deftrain_fn(parameters):passbench=Bench()bench.train('experiment_xyz',train_fn,parameters)
作者注释
<name>/crosses/
/001
├─ meta.yml
├─ history.pkl
└─ weights/
├─ 001.h5
└─ 002.h5
...
我写这个工具是为了自己使用你可以随意使用它。 如果你想看到任何额外的特点/报告存在的问题,请提交一个拉请求和/或打开一个问题。