Tensorboard not displaying all info in TF2 Object Detection API Training

AastaLLL · April 13, 2021, 11:35am

Hi,

Sorry for the late.
Based on the source, the summary do add in the eager training mode:

tensorflow/models/blob/master/research/object_detection/model_lib_v2.py#L311


      
          
          
detection_model._is_training = is_training  # pylint: disable=protected-access
          tf.keras.backend.set_learning_phase(is_training)
          
          
labels = model_lib.unstack_batch(
              labels, unpad_groundtruth_tensors=unpad_groundtruth_tensors)
          
          
with tf.GradientTape() as tape:
            losses_dict, _ = _compute_losses_and_predictions_dicts(
                detection_model, features, labels,
                training_step=training_step,
                add_regularization_loss=add_regularization_loss)
          
          
  losses_dict = normalize_dict(losses_dict, num_replicas)
          
          
trainable_variables = detection_model.trainable_variables
          
          
total_loss = losses_dict['Loss/total_loss']
          gradients = tape.gradient(total_loss, trainable_variables)
          
          
if clip_gradients_value:

Have you checked this issue with TensorFlow team?
They may know more about the difference in the summary API.

Thanks.

Topic		Replies	Views
TX2 Tensorflow 1.10 Training error Jetson TX2	10	1614	September 20, 2018
Overfitting models by using tensorboard TAO Toolkit	6	773	October 12, 2021
Integrating Tensorboard images into detectNet SSD training Jetson Nano tensorrt , tensorflow , jetson-inference	6	897	October 15, 2021
How to Visualizing the Retraining with TensorBoard Jetson TX2	11	1773	October 18, 2021
TensorFlow object detection and image classification accelerated for NVIDIA Jetson Jetson TX2	25	10546	June 3, 2019
Output file when training detectnet 2 TAO Toolkit	4	573	October 12, 2021
Missing TensorBoard support TAO Toolkit	4	1018	October 12, 2021
Tensorflow object detection api Frameworks (archived) tensorflow	0	361	May 4, 2020
Performance of Tensorflow (1.5) on Jetson TX2 slower than expected Jetson TX2	3	2797	October 18, 2021
Tensorflow 2 to TensorRT Jetson Nano tensorrt	2	656	October 15, 2021

Tensorboard not displaying all info in TF2 Object Detection API Training

Related topics