(This is a research project, not an official NVIDIA product.)
Milano (Machine learning autotuner and network optimizer) is a tool for enabling machine learning researchers and practitioners to perform massive hyperparameters and architecture searches.
You can use it to:
Your script can use any framework of your choice, for example, TensorFlow, PyTorch, Microsoft Cognitive Toolkit etc. or no framework at all. Milano only requires minimal changes to what your script accepts via command line and what it returns to stdout.
Currently supported backends:
We provide a script to convert the csv file output into two kinds of graphs:
To run the script, use:
python3 visualize.py --file [the name of the results csv file] --n [the number of samples to visualize] --subplots [the number of subplots to show in a plot] --max [the max value of benchmark you care about]