wubba lubba dub duband a Slack Channel 🚆
"Because it’s a jungle out there"
Trains is our solution to a problem we share with countless other researchers and developers in the machine learning/deep learning universe: Training production-grade deep learning models is a glorious but messy process. Trains tracks and controls the process by associating code version control, research projects, performance metrics, and model provenance.
We designed Trains specifically to require effortless integration so that teams can preserve their existing methods and practices. Use it on a daily basis to boost collaboration and visibility, or use it to automatically collect your experimentation logs, outputs, and data to one centralized server.
We have a demo server up and running at https://demoapp.trains.allegro.ai.
You can try out Trains and test your code, with no additional setup.
With only two lines of code, this is what you are getting:
Additionally, log data explicitly using Trains Explicit Logging.
Trains is a two part solution:
Trains python package auto-magically connects with your code
Trains requires only two lines of code for full integration.
To connect your code with Trains:
pip install trains
pip install trains[s3] pip install trains[gs] pip install trains[azure]
Add the following lines to your code
from trains import Task task = Task.init(project_name="my project", task_name="my task")
Run your code. When Trains connects to the server, a link is printed. For example
Trains Results page: https://demoapp.trains.allegro.ai/projects/76e5e2d45e914f52880621fe64601e85/experiments/241f06ae0f5c4b27b8ce8b64890ce152/output/log
Open the link and view your experiment parameters, model and tensorboard metrics
See examples here
We already have a demo server up and running for you at https://demoapp.trains.allegro.ai.
You can try out Trains without the need to install your own trains-server, just add the two lines of code, and it will automatically connect to the Trains demo-server.
Note that the demo server resets every 24 hours and all of the logged data is deleted.
When you are ready to use your own Trains server, go ahead and install trains-server.
Install and run trains-server (see Installing the Trains Server)
Run the initial configuration wizard for your Trains installation and follow the instructions to setup Trains package (http://trains-server-ip:port and user credentials)
After installing and configuring, you can access your configuration file at
Sample configuration file available here.
Trains is supported by the same team behind allegro.ai, where we build deep learning pipelines and infrastructure for enterprise companies.
We built Trains to track and control the glorious but messy process of training production-grade deep learning models. We are committed to vigorously supporting and expanding the capabilities of Trains.
We believe Trains is ground-breaking. We wish to establish new standards of experiment management in deep-learning and ML. Only the greater community can help us do that.
We promise to always be backwardly compatible. If you start working with Trains today, even though this project is currently in the beta stage, your logs and data will always upgrade with you.
Apache License, Version 2.0 (see the LICENSE for more information)
For feature requests or bug reports, please use GitHub issues.
Additionally, you can always find us at [email protected]
See the Trains Guidelines for Contributing.
May the force (and the goddess of learning rates) be with you!