Organize Machine Learning Experiments
Project description
LabML
LabML lets you monitor AI model training on mobile phones.
You can install this package using PIP.
pip install labml
To push to mobile website, you need obtain a token from web.lab-ml.com (Github lab-ml/app), and save statistics with tracker.save.
PyTorch example
from labml import tracker, experiment
with experiment.record(name='sample', exp_conf=conf, token: 'TOKEN from web.lab-ml.com'):
for i in range(50):
loss, accuracy = train()
tracker.save(i, {'loss': loss, 'accuracy': accuracy})
TensorFlow 2.X Keras example
from labml import experiment
from labml.utils.keras import LabMLKerasCallback
with experiment.record(name='sample', exp_conf=conf, token: 'TOKEN from web.lab-ml.com'):
for i in range(50):
model.fit(x_train, y_train, epochs=conf['epochs'], validation_data=(x_test, y_test),
callbacks=[LabMLKerasCallback()], verbose=None)
You can read the guides about creating an experiment, and saving statistics with tracker for details.
It automatically pushes data to Tensorboard, and you can keep your old experiments organized with the LabML Dashboard
All these software is 100% open source, and your logs will be stored locally for Tensorboard and LabML Dashboard. You will only be sending data away for web.lab-ml.com if you include a token url. This can also be locally installed.
LabML can also keep track of git commits, handle configurations, hyper-parameters, save and load checkpoints, and providing pretty logs.
We also have an API to create custom visualizations from artifacts and logs on Jupyter notebooks.
Links
Citing LabML
If you use LabML for academic research, please cite the library using the following BibTeX entry.
@misc{labml,
author = {Varuna Jayasiri, Nipun Wijerathne},
title = {LabML: A library to organize machine learning experiments},
year = {2020},
url = {https://lab-ml.com/},
}
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.