A tool for training deep learning cross validation training.
Project description
Trainbench
Trainbench is a tool for quickly setup a (Deep Learning) cross-validation training session with configuration instantiation, learned weights, plots, etc, recording at each training epoch.
Compatible with Keras.
⚠️ This package is currently under active development!
Getting started
Install
pip install trainbench
Create a train session
# train.py
# name
name = 'experiment_01'
# parameters to cross validate against
parameters = {
'fc_size': [256, 512, 1024]
# ...
}
def train(parameters):
xyz = parameters['xyz']
# ...
Running from the command line
trainbench .
Checkout your results
<name>/crosses/
/001
├─ meta.yml
├─ history.pkl
└─ weights/
├─ 001.h5
└─ 002.h5
...
In Jupyter Notebooks/Python script
from trainbench import Bench
parameters = {
}
def train_fn(parameters):
pass
bench = Bench()
bench.train('experiment_xyz', train_fn, parameters)
Author notes
I'm wrote this tool for my own usage. Feel free to use it at your own will. If you would like to see any additional features / report existing issues please submit a pull request and/or open an issue.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
trainbench-0.0.2.tar.gz
(3.1 kB
view hashes)