ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 17:17:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
62bc54d7be
clearml/docs/brief.md
allegroai a0ecc6d516 Documentation
2019-07-08 23:29:09 +03:00

3.7 KiB
Raw Blame History

What is TRAINS?

Behind every great scientist are great repeatable methods. Sadly, this is easier said than done.

When talented scientists, engineers, or developers work on their own, a mess may be unavoidable. Yet, it may still be manageable. However, with time and more people joining your project, managing the clutter takes its toll on productivity. As your project moves toward production, visibility and provenance for scaling your deep-learning efforts are a must.

For teams or entire companies, TRAINS logs everything in one central server and takes on the responsibilities for visibility and provenance so productivity does not suffer. TRAINS records and manages various deep learning research workloads and does so with practically zero integration costs.

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.

Main Features

  • Integrate with your current work flow with minimal effort
    • Seamless integration with leading frameworks, including: PyTorch, TensorFlow, Keras, and others coming soon
    • Support for Jupyter Notebook (see trains-jupyter-plugin) and PyCharm remote debugging (see trains-pycharm-plugin)
  • Log everything. Experiments become truly repeatable
    • Model logging with automatic association of model + code + parameters + initial weights
    • Automatically create a copy of models on centralized storage (supports shared folders, S3, GS, and Azure is coming soon!)
  • Share and collaborate
    • Multi-user process tracking and collaboration
    • Centralized server for aggregating logs, records, and general bookkeeping
  • Increase productivity
    • Comprehensive experiment comparison: code commits, initial weights, hyper-parameters and metric results
  • Order & Organization
    • Manage and organize your experiments in projects
    • Query capabilities; sort and filter experiments by results metrics
  • And more
    • Stop an experiment on a remote machine using the web-app
    • A field-tested, feature-rich SDK for your on-the-fly customization needs

TRAINS Automatically Logs

  • Git repository, branch, commit id and entry point (git diff coming soon)
    • Hyper-parameters, including
    • ArgParser for command line parameters with currently used values
    • Tensorflow Defines (absl-py)
  • Explicit parameters dictionary
  • Initial model weights file
  • Model snapshots
  • stdout and stderr
  • Tensorboard/TensorboardX scalars, metrics, histograms, images (with audio coming soon)
  • Matplotlib

How TRAINS Works

TRAINS is a two part solution:

  1. TRAINS python package (auto-magically connects your code, see Using TRAINS)
  2. TRAINS-server for logging, querying, control and UI (Web-App)

The following diagram illustrates the interaction of the TRAINS-server and a GPU training machine using the TRAINS python package