Tensorboard for MaskRCNN TAO model

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

Please provide the following information when requesting support.

• Hardware: NVIDIA GeForce RTX 2080 Ti
• Network Type: Mask_rcnn
• TLT Version toolkit_version: 4.0.1
• Training spec file maskrcnn_train_resnet50.txt (2.1 KB)
• How to reproduce the issue ?

Hi all,

I am trying to train a MaskRCNN model using the TAO toolkit, and monitor the progress with Tensorboard.
This page says I should add the following section to the MaskRCNN spec file, which I have uploaded:
training_config { visualizer { enabled: true } }
However, this page does not mention such an action for this network, even though this page says it should be supported.
Ultimately, tao mask_rcnn train throws the following error as training is about to start: google.protobuf.text_format.ParseError: 17:1 : Message type "Experiment" has no field named "train_config".

Is or will Tensorboard be supported for MaskRCNN? Is there something I am doing wrong?

Kind Regards,

2

Please add below directly

visualize_images_summary: true

under
num_examples_per_epoch: 100

3

Thank you for the quick reply.
This worked fine!

4

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.