Package: tensorwatch
Welcome to TensorWatchTensorWatch is a debugging and visualization tool designed for data science, deep learning and reinforcement learning from Microsoft Research. It works in Jupyter Notebook to show real-time visualizations of your machine learning training and perform several other key analysis tasks for your models and data.TensorWatch is designed to be flexible and extensible so you can also build your own custom visualizations, UIs, and dashboards. Besides traditional "what-you-see-is-what-you-log" approach, it also has a unique capability to execute arbitrary queries against your live ML training process, return a stream as a result of the query and view this stream using your choice of a visualizer (we call thisLazy Logging Mode).TensorWatch is under heavy development with a goal of providing a platform for debugging machine learning in one easy to use, extensible, and hackable package.How to Get Itpip install tensorwatchTensorWatch supports Python 3.x and is tested with PyTorch 0.4-1.x. Most features should also work with TensorFlow eager tensors. TensorWatch uses graphviz to create network diagrams and depending on your platform sometime you might need to manuallyinstallit.How to Use ItQuick StartHere's simple code that logs an integer and its square as a tuple every second to TensorWatch:import tensorwatch as tw
import time

# streams will be stored in test.log file
w = tw.Watcher(filename='test.log')

# create a stream for logging
s = w.create_stream(name='metric1')

# generate Jupyter Notebook to view real-time streams
w.make_notebook()

for i in range(1000):
    # write x,y pair we want to log
    s.write((i, i*i))

    time.sleep(1)When you run this code, you will notice a Jupyter Notebook filetest.ipynbgets created in your script folder. From a command prompt typejupyter notebookand selecttest.ipynb. ChooseCell > Run allin the menu to see the real-time line graph as values get written in your script.Here's the output you will see in Jupyter Notebook:To dive deeper into the various other features, please seeTutorialsandnotebooks.How does this work?When you write to a TensorWatch stream, the values get serialized and sent to a TCP/IP socket as well as the file you specified. From Jupyter Notebook, we load the previously logged values from the file and then listen to that TCP/IP socket for any future values. The visualizer listens to the stream and renders the values as they arrive.Ok, so that's a very simplified description. The TensorWatch architecture is actually much more powerful. Almost everything in TensorWatch is astream. Files, sockets, consoles and even visualizers are streams themselves. A cool thing about TensorWatch streams is that they can listen to any other streams. This allows TensorWatch to create adata flow graph. This means that a visualizer can listen to many streams simultaneously, each of which could be a file, a socket or some other stream. You can recursively extend this to build arbitrary data flow graphs. TensorWatch decouples streams from how they get stored and how they get visualized.VisualizationsIn the above example, the line graph is used as the default visualization. However, TensorWatch supports many other diagram types including histograms, pie charts, scatter charts, bar charts and 3D versions of many of these plots. You can log your data, specify the chart type you want and let TensorWatch take care of the rest.One of the significant strengths of TensorWatch is the ability to combine, compose, and create custom visualizations effortlessly. For example, you can choose to visualize an arbitrary number of streams in the same plot. Or you can visualize the same stream in many different plotssimultaneously. Or you can place an arbitrary set of visualizations side-by-side. You can even create your own custom visualization widget simply by creating a new Python class, implementing a few methods.Comparing Results of Multiple RunsEach TensorWatch stream may contain a metric of your choice. By default, TensorWatch saves all streams in a single file, but you could also choose to save each stream in separate files or not to save them at all (for example, sending streams over sockets or into the console directly, zero hit to disk!). Later you can open these streams and direct them to one or more visualizations. This design allows you to quickly compare the results from your different experiments in your choice of visualizations easily.Training within Jupyter NotebookOften you might prefer to do data analysis, ML training, and testing - all from within Jupyter Notebook instead of from a separate script. TensorWatch can help you do sophisticated, real-time visualizations effortlessly from code that is run within a Jupyter Notebook end-to-end.Lazy Logging ModeA unique feature in TensorWatch is the ability to query the live running process, retrieve the result of this query as a stream and direct this stream to your preferred visualization(s). You don't need to log any data beforehand. We call this new way of debugging and visualization alazy logging mode.For example, as seen below, we visualize input and output image pairs, sampled randomly during the training of an autoencoder on a fruits dataset. These images were not logged beforehand in the script. Instead, the user sends query as a Python lambda expression which results in a stream of images that gets displayed in the Jupyter Notebook:SeeLazy Logging Tutorial.Pre-Training and Post-Training TasksTensorWatch leverages several excellent libraries includinghiddenlayer,torchstat,Visual Attributionto allow performing the usual debugging and analysis activities in one consistent package and interface.For example, you can view the model graph with tensor shapes with a one-liner:You can view statistics for different layers such as flops, number of parameters, etc:See notebook.You can view the dataset in a lower dimensional space using techniques such as t-SNE:See notebook.Prediction ExplanationsWe wish to provide various tools for explaining predictions to help debugging models. Currently, we offer several explainers for convolutional networks, includingLime. For example, the following highlights the areas that cause the Resnet50 model to make a prediction for class 240 for the Imagenet dataset:See notebook.TutorialsSimple Logging TutorialLazy Logging TutorialUsing TensorWatch for Deep Learning Training (MNIST)Using TensorWatch for Deep Learning Training (Food360)Exploring Data Using T-SNEPredication Explainers for Convolutional Neural NetworksViewing Model Graph and StatisticsPaperMore technical details are available inTensorWatch paper (EICS 2019 Conference). Please cite this as:@inproceedings{tensorwatch2019eics,
  author    = {Shital Shah and Roland Fernandez and Steven M. Drucker},
  title     = {A system for real-time interactive analysis of deep learning training},
  booktitle = {Proceedings of the {ACM} {SIGCHI} Symposium on Engineering Interactive
               Computing Systems, {EICS} 2019, Valencia, Spain, June 18-21, 2019},
  pages     = {16:1--16:6},
  year      = {2019},
  crossref  = {DBLP:conf/eics/2019},
  url       = {https://arxiv.org/abs/2001.01215},
  doi       = {10.1145/3319499.3328231},
  timestamp = {Fri, 31 May 2019 08:40:31 +0200},
  biburl    = {https://dblp.org/rec/bib/conf/eics/ShahFD19},
  bibsource = {dblp computer science bibliography, https://dblp.org}
}ContributeWe would love your contributions, feedback, questions, and feature requests! Pleasefile a Github issueor send us a pull request. Please review theMicrosoft Code of Conductandlearn more.ContactJoin the TensorWatch group onFacebookto stay up to date or ask any questions.CreditsTensorWatch utilizes several open source libraries for many of its features. These includes:hiddenlayer,torchstat,Visual-Attribution,pyzmq,receptivefield,nbformat. Please seeinstall_requiressection insetup.pyfor upto date list.LicenseThis project is released under the MIT License. Please review theLicense filefor more details.