Skip to content

GitLab

Commit 31c3ab9e authored by vishnubanna's avatar vishnubanna
Browse files

pr comment to explain decoder

parent 4af9988d
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
official/vision/beta/projects/yolo/tasks/image_classification.py
View file @ 31c3ab9e
...@@ -83,7 +83,6 @@ class ImageClassificationTask(image_classification.ImageClassificationTask): ...@@ -83,7 +83,6 @@ class ImageClassificationTask(image_classification.ImageClassificationTask):
model_outputs=outputs, labels=labels, aux_losses=model.losses) model_outputs=outputs, labels=labels, aux_losses=model.losses)
# Scales loss as the default gradients allreduce performs sum inside the # Scales loss as the default gradients allreduce performs sum inside the
# optimizer. # optimizer.
# tf.print("training loss: ", loss, end="\r")
scaled_loss = loss / num_replicas scaled_loss = loss / num_replicas
# For mixed_precision policy, when LossScaleOptimizer is used, loss is # For mixed_precision policy, when LossScaleOptimizer is used, loss is
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment