Visualize Trial with TensorBoard¶
You can launch a tensorboard process cross one or multi trials within webportal since NNI v2.2. This feature supports local training service and reuse mode training service with shared storage for now, and will support more scenarios in later nni version.
Make sure tensorboard installed in your environment. If you never used tensorboard, here are getting start tutorials for your reference, tensorboard with tensorflow, tensorboard with pytorch.
Use WebUI Launch Tensorboard¶
NNI will automatically fetch the
tensorboard subfolder under trial’s output folder as tensorboard logdir. So in trial’s source code, you need to save the tensorboard logs under
NNI_OUTPUT_DIR/tensorboard. This log path can be joined as:
log_dir = os.path.join(os.environ["NNI_OUTPUT_DIR"], 'tensorboard')
Like compare, select the trials you want to combine to launch the tensorboard at first, then click the
After click the
OKbutton in the pop-up box, you will jump to the tensorboard portal.
You can see the
SequenceID-TrialIDon the tensorboard portal.
If you want to open the portal you have already launched, click the tensorboard id. If you don’t need the tensorboard anymore, click
Stop all tensorboard button.