Remove transformation in light of ponder-lab/Hybridize-Functions-Refa…

…ctoring#363. This function will not be transformed.
ponder-lab · Aug 23, 2024 · 6a6420e · 6a6420e
1 parent 45bdb5a
commit 6a6420e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/4-Object_Detection/RPN/train.py b/4-Object_Detection/RPN/train.py
@@ -98,7 +98,6 @@ def DataGenerator(synthetic_dataset_path, batch_size):
             target_masks[i]  = target[2]
         yield images, target_scores, target_bboxes, target_masks
 
-@tf.function
 def compute_loss(target_scores, target_bboxes, target_masks, pred_scores, pred_bboxes):
     """
     target_scores shape: [1, 45, 60, 9, 2],  pred_scores shape: [1, 45, 60, 9, 2]