tf-models-nightly 2.20.0.dev20251106__py2.py3-none-any.whl → 2.20.0.dev20260109__py2.py3-none-any.whl

official/nlp/modeling/layers/transformer_encoder_block.py

@@ -121,6 +121,7 @@ class TransformerEncoderBlock(tf_keras.layers.Layer):
       lowrank_query_seq_proj_dim=None,
       enable_talking_heads=False,
       enable_gqa_optimization=False,
+      softmax_robust_masking=False,
       **kwargs,
   ):
     """Initializes `TransformerEncoderBlock`.
@@ -209,6 +210,8 @@ class TransformerEncoderBlock(tf_keras.layers.Layer):
         https://arxiv.org/pdf/2003.02436.
       enable_gqa_optimization: Enable GQA optimization in multi-query attention.
         This flag is valid only when num_kv_heads is set for GQA.
+      softmax_robust_masking: If true, will use a more numerically robust
+        masking impl for softmax.
       **kwargs: keyword arguments.
     """
     util.filter_kwargs(kwargs)
@@ -253,6 +256,7 @@ class TransformerEncoderBlock(tf_keras.layers.Layer):
     self._lowrank_query_seq_proj_dim = lowrank_query_seq_proj_dim
     self._enable_talking_heads = enable_talking_heads
     self._enable_gqa_optimization = enable_gqa_optimization
+    self._softmax_robust_masking = softmax_robust_masking
     if (
         self._src_block_size is not None
         and self._num_kv_heads is not None
@@ -314,6 +318,7 @@ class TransformerEncoderBlock(tf_keras.layers.Layer):
         bias_initializer=tf_utils.clone_initializer(self._bias_initializer),
         attention_axes=self._attention_axes,
         output_shape=self._output_last_dim,
+        softmax_robust_masking=self._softmax_robust_masking,
         name="self_attention",
     )
     common_kwargs = dict(
@@ -512,6 +517,7 @@ class TransformerEncoderBlock(tf_keras.layers.Layer):
         "linformer_dim": self._linformer_dim,
         "linformer_shared_kv_projection": self._linformer_shared_kv_projection,
         "lowrank_query_seq_proj_dim": self._lowrank_query_seq_proj_dim,
+        "softmax_robust_masking": self._softmax_robust_masking
     }
     base_config = super().get_config()
     return dict(list(base_config.items()) + list(config.items()))

official/vision/data/fake_feature_generator.py

@@ -28,7 +28,8 @@ def generate_image_np(height: int,
   """Returns a fake numpy image matrix array."""
   return np.reshape(
       np.mod(np.arange(height * width * num_channels), 255).astype(np.uint8),
-      newshape=(height, width, num_channels))
+      (height, width, num_channels),
+  )
 
 
 def generate_normalized_boxes_np(num_boxes: int) -> np.ndarray:
@@ -82,8 +83,7 @@ def generate_instance_masks_np(height: int,
   box_heights = boxes_np[:, 3].astype(int) - ymins
 
   for i, (x, y, w, h) in enumerate(zip(xmins, ymins, box_widths, box_heights)):
-    instance_masks_np[i, y:y + h, x:x + w, :] = np.reshape(
-        np.mod(np.arange(h * w), 2).astype(np.uint8), newshape=(h, w, 1))
+    instance_masks_np[i, y:y + h, x:x + w, :] = np.reshape(np.mod(np.arange(h * w), 2).astype(np.uint8), (h, w, 1))
   return instance_masks_np
 
 

official/vision/evaluation/coco_utils.py

@@ -241,7 +241,7 @@ def convert_groundtruths_to_coco_dataset(groundtruths, label_map=None):
             'num_groundtruths is larger than max_num_instances, %d v.s. %d',
             num_instances, max_num_instances)
         num_instances = max_num_instances
-      for k in range(int(num_instances)):
+      for k in range(int(np.squeeze(num_instances))):
         ann = {}
         ann['image_id'] = int(groundtruths['source_id'][i][j])
         if 'is_crowds' in groundtruths:

{tf_models_nightly-2.20.0.dev20251106.dist-info → tf_models_nightly-2.20.0.dev20260109.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tf-models-nightly
-Version: 2.20.0.dev20251106
+Version: 2.20.0.dev20260109
 Summary: TensorFlow Official Models
 Home-page: https://github.com/tensorflow/models
 Author: Google Inc.

{tf_models_nightly-2.20.0.dev20251106.dist-info → tf_models_nightly-2.20.0.dev20260109.dist-info}/RECORD

@@ -363,7 +363,7 @@ official/nlp/modeling/layers/tn_expand_condense_test.py,sha256=QWq1dJqQUPe5n69K3
 official/nlp/modeling/layers/tn_transformer_expand_condense.py,sha256=omzTkCBEk2TOkHEYDEBwve6WsOitX7IIJHzeKXdqDq0,11012
 official/nlp/modeling/layers/tn_transformer_test.py,sha256=pSCONEZRI4J9_6QLTJ3g_ynUYLrRXsJ1c2YMSiOV_60,8893
 official/nlp/modeling/layers/transformer.py,sha256=VjUO-gVj_PnavbT_vSrg5NDKMr0SRSiqSg5ktd42m5M,20087
-official/nlp/modeling/layers/transformer_encoder_block.py,sha256=BiL8ErBs-m0UZ6ONVJV0ncfWX3LhMhPetIhfH2VvuD4,29910
+official/nlp/modeling/layers/transformer_encoder_block.py,sha256=78nIg3hVFKLeqJ-Hdg4j1scfskZAnm2CSHqlTeoSRcc,30236
 official/nlp/modeling/layers/transformer_encoder_block_test.py,sha256=g7oMDPvwg6Fv75SBdm6BInXPI8r5GcItBRjLFGuObyg,37821
 official/nlp/modeling/layers/transformer_scaffold.py,sha256=qmzhCJvbbFVF9zDqnfO4Zs2JDXwKhK7iEBOhsU6-KpQ,15704
 official/nlp/modeling/layers/transformer_scaffold_test.py,sha256=dRJwesTBKm-mF5mDHrHfVpVNnxa-Wx-fj_4ZHDPTpE0,19920
@@ -1013,7 +1013,7 @@ official/vision/configs/video_classification.py,sha256=GIdceuHsXTI3KdAbh4tPnctEk
 official/vision/configs/video_classification_test.py,sha256=pLbpLcN4HqKtFXVuAbR1iSVWWZ3V_cImNSzQb6E86Zw,1879
 official/vision/data/__init__.py,sha256=atDqjsx74qLrX5FY_wyWi2TL8i8ZELZRsTU9c-sx51I,609
 official/vision/data/create_coco_tf_record.py,sha256=pb3d4yAR7xy3nbR93MFadDKlUBgJKvtwmMyyNa4y1Bs,22742
-official/vision/data/fake_feature_generator.py,sha256=4vPWaEqpW8ws8AfluWLhuORSx23j6rKDQvuBw1YChWc,5109
+official/vision/data/fake_feature_generator.py,sha256=9VvarI-2wYIS5YPR5qN-V4FGgWHzT_25T0CA8-Dw5Xs,5086
 official/vision/data/image_utils.py,sha256=3aSu4vYGYYHwyVwqaR_9TGH3crJVxTjC6VXuT5S8AtA,3503
 official/vision/data/image_utils_test.py,sha256=Fp_QksSDMgwH8QiXWvcE_XYDsWYdQpha1wOz1DBzl1s,4071
 official/vision/data/process_coco_few_shot_json_files.py,sha256=H3hbzbSS4nPXem5zE1rObesW73jJ_pVBDr0CMgtvubg,6061
@@ -1047,7 +1047,7 @@ official/vision/dataloaders/video_input.py,sha256=yp-3VSuHLbshwOKZWTSWHOP9NoKitA
 official/vision/dataloaders/video_input_test.py,sha256=ZQWkjK5tpzqU3S_pN3GECfPBJ8pIm_IV-WEQkqtifr4,7532
 official/vision/evaluation/__init__.py,sha256=atDqjsx74qLrX5FY_wyWi2TL8i8ZELZRsTU9c-sx51I,609
 official/vision/evaluation/coco_evaluator.py,sha256=23wnGNUZB4Tb01Y6fyjxcfgu7MpUsgaEjrMMTbrMMRc,15582
-official/vision/evaluation/coco_utils.py,sha256=mN8Dz_WBIgvrGBhb5PJte0qTW09mXjL-PwIuE4F_nwU,17860
+official/vision/evaluation/coco_utils.py,sha256=uCbRRe78-ZNKNss2V3WejHTfOO3dj2zzrcYWzZzgmns,17872
 official/vision/evaluation/coco_utils_test.py,sha256=B_8jj1YkgipPOr3BzNrwXjXgVOt-_k6kesCFB5kkxDA,3199
 official/vision/evaluation/instance_metrics.py,sha256=m61_ApFqCwOGsbGtXa0NGMxlLYyx_cqcUQDPOPRpdWM,29114
 official/vision/evaluation/instance_metrics_test.py,sha256=J9Jf4_Oc2ULJF9pwD3ikO1Wiebq4GB4dbWSr3Lxm16I,10823
@@ -1248,9 +1248,9 @@ tensorflow_models/tensorflow_models_test.py,sha256=yiAneltAW3NHSj3fUSvHNBjfq0MGZ
 tensorflow_models/nlp/__init__.py,sha256=8uQd4wI6Zc4IJMPjtQifMeWVbPFkTxqYh66wfivCOL4,807
 tensorflow_models/uplift/__init__.py,sha256=NzaweFf4ZmhRb2l_fuV6bP-2N8oSO3xu6xJqVb1UmpY,999
 tensorflow_models/vision/__init__.py,sha256=ks420Ooqzi0hU7HnQpM5rylLaE-YcJdJkBx_umVaXlE,833
-tf_models_nightly-2.20.0.dev20251106.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
-tf_models_nightly-2.20.0.dev20251106.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
-tf_models_nightly-2.20.0.dev20251106.dist-info/METADATA,sha256=hAiVT3AJkJAsf2opEbBOBlWtg54F_t6-gNQobBmNW7A,1432
-tf_models_nightly-2.20.0.dev20251106.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
-tf_models_nightly-2.20.0.dev20251106.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
-tf_models_nightly-2.20.0.dev20251106.dist-info/RECORD,,
+tf_models_nightly-2.20.0.dev20260109.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
+tf_models_nightly-2.20.0.dev20260109.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
+tf_models_nightly-2.20.0.dev20260109.dist-info/METADATA,sha256=kySce29gd90FHA2kt9WkRiZsB76xRkwQ30wvkbm43BM,1432
+tf_models_nightly-2.20.0.dev20260109.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
+tf_models_nightly-2.20.0.dev20260109.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
+tf_models_nightly-2.20.0.dev20260109.dist-info/RECORD,,