tf-models-nightly 2.19.0.dev20250129__py2.py3-none-any.whl → 2.19.0.dev20250130__py2.py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- official/common/distribute_utils_test.py +12 -11
- official/recommendation/data_test.py +3 -3
- official/utils/misc/model_helpers_test.py +2 -2
- {tf_models_nightly-2.19.0.dev20250129.dist-info → tf_models_nightly-2.19.0.dev20250130.dist-info}/METADATA +1 -1
- {tf_models_nightly-2.19.0.dev20250129.dist-info → tf_models_nightly-2.19.0.dev20250130.dist-info}/RECORD +9 -9
- {tf_models_nightly-2.19.0.dev20250129.dist-info → tf_models_nightly-2.19.0.dev20250130.dist-info}/AUTHORS +0 -0
- {tf_models_nightly-2.19.0.dev20250129.dist-info → tf_models_nightly-2.19.0.dev20250130.dist-info}/LICENSE +0 -0
- {tf_models_nightly-2.19.0.dev20250129.dist-info → tf_models_nightly-2.19.0.dev20250130.dist-info}/WHEEL +0 -0
- {tf_models_nightly-2.19.0.dev20250129.dist-info → tf_models_nightly-2.19.0.dev20250130.dist-info}/top_level.txt +0 -0
@@ -43,14 +43,14 @@ class DistributeUtilsTest(tf.test.TestCase):
|
|
43
43
|
|
44
44
|
def test_one_device_strategy_cpu(self):
|
45
45
|
ds = distribute_utils.get_distribution_strategy('one_device', num_gpus=0)
|
46
|
-
self.
|
47
|
-
self.
|
46
|
+
self.assertEqual(ds.num_replicas_in_sync, 1)
|
47
|
+
self.assertEqual(len(ds.extended.worker_devices), 1)
|
48
48
|
self.assertIn('CPU', ds.extended.worker_devices[0])
|
49
49
|
|
50
50
|
def test_one_device_strategy_gpu(self):
|
51
51
|
ds = distribute_utils.get_distribution_strategy('one_device', num_gpus=1)
|
52
|
-
self.
|
53
|
-
self.
|
52
|
+
self.assertEqual(ds.num_replicas_in_sync, 1)
|
53
|
+
self.assertEqual(len(ds.extended.worker_devices), 1)
|
54
54
|
self.assertIn('GPU', ds.extended.worker_devices[0])
|
55
55
|
|
56
56
|
def test_mirrored_strategy(self):
|
@@ -58,8 +58,8 @@ class DistributeUtilsTest(tf.test.TestCase):
|
|
58
58
|
_ = distribute_utils.get_distribution_strategy(num_gpus=0)
|
59
59
|
# 5 GPUs.
|
60
60
|
ds = distribute_utils.get_distribution_strategy(num_gpus=5)
|
61
|
-
self.
|
62
|
-
self.
|
61
|
+
self.assertEqual(ds.num_replicas_in_sync, 5)
|
62
|
+
self.assertEqual(len(ds.extended.worker_devices), 5)
|
63
63
|
for device in ds.extended.worker_devices:
|
64
64
|
self.assertIn('GPU', device)
|
65
65
|
|
@@ -105,12 +105,13 @@ class DistributeUtilsTest(tf.test.TestCase):
|
|
105
105
|
ds, tf.distribute.TPUStrategy)
|
106
106
|
|
107
107
|
def test_invalid_strategy(self):
|
108
|
-
with self.
|
109
|
-
ValueError,
|
110
|
-
|
108
|
+
with self.assertRaisesRegex(
|
109
|
+
ValueError, 'distribution_strategy must be a string but got: False. If'
|
110
|
+
):
|
111
111
|
distribute_utils.get_distribution_strategy(False)
|
112
|
-
with self.
|
113
|
-
ValueError, 'distribution_strategy must be a string but got: 1'
|
112
|
+
with self.assertRaisesRegex(
|
113
|
+
ValueError, 'distribution_strategy must be a string but got: 1'
|
114
|
+
):
|
114
115
|
distribute_utils.get_distribution_strategy(1)
|
115
116
|
|
116
117
|
def test_get_strategy_scope(self):
|
@@ -190,7 +190,7 @@ class BaseTest(tf.test.TestCase):
|
|
190
190
|
train_examples[l].add((u_raw, i_raw))
|
191
191
|
counts[(u_raw, i_raw)] += 1
|
192
192
|
|
193
|
-
self.
|
193
|
+
self.assertRegex(md5.hexdigest(), END_TO_END_TRAIN_MD5)
|
194
194
|
|
195
195
|
num_positives_seen = len(train_examples[True])
|
196
196
|
self.assertEqual(producer._train_pos_users.shape[0], num_positives_seen)
|
@@ -254,7 +254,7 @@ class BaseTest(tf.test.TestCase):
|
|
254
254
|
# from the negatives.
|
255
255
|
assert (u_raw, i_raw) not in self.seen_pairs
|
256
256
|
|
257
|
-
self.
|
257
|
+
self.assertRegex(md5.hexdigest(), END_TO_END_EVAL_MD5)
|
258
258
|
|
259
259
|
def _test_fresh_randomness(self, constructor_type):
|
260
260
|
train_epochs = 5
|
@@ -300,7 +300,7 @@ class BaseTest(tf.test.TestCase):
|
|
300
300
|
else:
|
301
301
|
negative_counts[(u, i)] += 1
|
302
302
|
|
303
|
-
self.
|
303
|
+
self.assertRegex(md5.hexdigest(), FRESH_RANDOMNESS_MD5)
|
304
304
|
|
305
305
|
# The positive examples should appear exactly once each epoch
|
306
306
|
self.assertAllEqual(
|
@@ -82,7 +82,7 @@ class SyntheticDataTest(tf.test.TestCase):
|
|
82
82
|
for n in range(5):
|
83
83
|
inp, lab = sess.run((input_element, label_element))
|
84
84
|
self.assertAllClose(inp, [123., 123., 123., 123., 123.])
|
85
|
-
self.
|
85
|
+
self.assertEqual(lab, 456)
|
86
86
|
|
87
87
|
def test_generate_only_input_data(self):
|
88
88
|
d = model_helpers.generate_synthetic_data(
|
@@ -111,7 +111,7 @@ class SyntheticDataTest(tf.test.TestCase):
|
|
111
111
|
element = tf.compat.v1.data.make_one_shot_iterator(d).get_next()
|
112
112
|
self.assertIn('a', element)
|
113
113
|
self.assertIn('b', element)
|
114
|
-
self.
|
114
|
+
self.assertEqual(len(element['b']), 2)
|
115
115
|
self.assertIn('c', element['b'])
|
116
116
|
self.assertIn('d', element['b'])
|
117
117
|
self.assertNotIn('c', element)
|
@@ -2,7 +2,7 @@ official/__init__.py,sha256=7oiypy0N82PDw9aSdcJBLVoGTd_oRSUOdvuJhMv4leQ,609
|
|
2
2
|
official/common/__init__.py,sha256=Yqe5TjLrAR68lQ5G781dwlbbT6mELRByDipCtr4jQY4,610
|
3
3
|
official/common/dataset_fn.py,sha256=MSRNhNGvMnZtjkuxFIZ90vGJgKlr2JeGaEYpz82Y3Ew,1858
|
4
4
|
official/common/distribute_utils.py,sha256=dEciUuyBsxfv6T9rvPk_5zcvLyDFv0lPkhkFvyOGe_0,9575
|
5
|
-
official/common/distribute_utils_test.py,sha256=
|
5
|
+
official/common/distribute_utils_test.py,sha256=KSJ8zhiWrbmXzgoNC86gbwXGjnjoxtxou4S3_VFyilw,4908
|
6
6
|
official/common/flags.py,sha256=rtCyljUH5M0p5a1CvBnj6tdubZe4XkwzqiOz1ML6F7I,5347
|
7
7
|
official/common/registry_imports.py,sha256=ez-j0sOvVyYBHHGM5hNjTF0jqHUbHI0zv-4SHe_AZ4s,843
|
8
8
|
official/common/streamz_counters.py,sha256=OtdlnI5aosH_DWP8jx2MGEpjs5AhyMztr5E5qhrC4nE,1057
|
@@ -902,7 +902,7 @@ official/recommendation/constants.py,sha256=gCMNNJHwX0fUEabCiTJa7tV7KHoXD5wc-jMI
|
|
902
902
|
official/recommendation/create_ncf_data.py,sha256=_UEOulLVbZLY1MN8FukpD0hEN5n030CpeWpUu46gK5I,3999
|
903
903
|
official/recommendation/data_pipeline.py,sha256=lE9dc5zLgwXN8wohNTGTClUvFytQSqbmU-Tv0jQ3PMU,37216
|
904
904
|
official/recommendation/data_preprocessing.py,sha256=haYd4hoEgTLEfgTDzS037Mbk-IluQzr6d-4RrdyDsNQ,10318
|
905
|
-
official/recommendation/data_test.py,sha256=
|
905
|
+
official/recommendation/data_test.py,sha256=fkPAn16YK3z3UWQRDrKOlLSvVXn2sWKSCM7-R_cMm_s,12807
|
906
906
|
official/recommendation/movielens.py,sha256=vET5x3zc8vW9ruZatMO3gRFUdPJjeDFJtHXfWr0bOAI,9725
|
907
907
|
official/recommendation/ncf_common.py,sha256=r9P3HH4JQl4Tu3ae3RnNrdSa2ZcH7N7ZXyxCMhJwqY0,12288
|
908
908
|
official/recommendation/ncf_input_pipeline.py,sha256=y2DCc5IVBK3LBk9yUVWfwb2dz-DCt2T0suASZI02UxU,6974
|
@@ -988,7 +988,7 @@ official/utils/flags/flags_test.py,sha256=UFGiGEjyR1wCvbqjdCxhzjGWm0xLrc4kn-zkaE
|
|
988
988
|
official/utils/misc/__init__.py,sha256=7oiypy0N82PDw9aSdcJBLVoGTd_oRSUOdvuJhMv4leQ,609
|
989
989
|
official/utils/misc/keras_utils.py,sha256=AxsatfCV3Desrp48_WGxoQsfMO0uA9bcEXFXa7djRqs,7793
|
990
990
|
official/utils/misc/model_helpers.py,sha256=E7AHQ-3ppDrtmKUFKAvaEnBAZI2TW3p8Hg8-I4Rvp3w,3370
|
991
|
-
official/utils/misc/model_helpers_test.py,sha256=
|
991
|
+
official/utils/misc/model_helpers_test.py,sha256=40paIH7IzeFa1OCSXUSmfMDAJYTYj3C2BDAR1t1b9fU,4557
|
992
992
|
official/utils/testing/__init__.py,sha256=7oiypy0N82PDw9aSdcJBLVoGTd_oRSUOdvuJhMv4leQ,609
|
993
993
|
official/utils/testing/integration.py,sha256=qo00sSG1fC65wI5nVZ6qm2Iek8CBShpMODgt2IdjPsE,2220
|
994
994
|
official/utils/testing/mock_task.py,sha256=Lag4UZpJlhHu0nZy2HG-7Rcnkrg0Xv6mW5zj18_a7Ek,3312
|
@@ -1248,9 +1248,9 @@ tensorflow_models/tensorflow_models_test.py,sha256=nc6A9K53OGqF25xN5St8EiWvdVbda
|
|
1248
1248
|
tensorflow_models/nlp/__init__.py,sha256=4tA5Pf4qaFwT-fIFOpX7x7FHJpnyJT-5UgOeFYTyMlc,807
|
1249
1249
|
tensorflow_models/uplift/__init__.py,sha256=mqfa55gweOdpKoaQyid4A_4u7xw__FcQeSIF0k_pYmI,999
|
1250
1250
|
tensorflow_models/vision/__init__.py,sha256=zBorY_v5xva1uI-qxhZO3Qh-Dii-Suq6wEYh6hKHDfc,833
|
1251
|
-
tf_models_nightly-2.19.0.
|
1252
|
-
tf_models_nightly-2.19.0.
|
1253
|
-
tf_models_nightly-2.19.0.
|
1254
|
-
tf_models_nightly-2.19.0.
|
1255
|
-
tf_models_nightly-2.19.0.
|
1256
|
-
tf_models_nightly-2.19.0.
|
1251
|
+
tf_models_nightly-2.19.0.dev20250130.dist-info/AUTHORS,sha256=1dG3fXVu9jlo7bul8xuix5F5vOnczMk7_yWn4y70uw0,337
|
1252
|
+
tf_models_nightly-2.19.0.dev20250130.dist-info/LICENSE,sha256=WxeBS_DejPZQabxtfMOM_xn8qoZNJDQjrT7z2wG1I4U,11512
|
1253
|
+
tf_models_nightly-2.19.0.dev20250130.dist-info/METADATA,sha256=cg6liQ_6KJHd79FTLdmOpyKuhFRPxYOago3WhfX7_rQ,1432
|
1254
|
+
tf_models_nightly-2.19.0.dev20250130.dist-info/WHEEL,sha256=kGT74LWyRUZrL4VgLh6_g12IeVl_9u9ZVhadrgXZUEY,110
|
1255
|
+
tf_models_nightly-2.19.0.dev20250130.dist-info/top_level.txt,sha256=gum2FfO5R4cvjl2-QtP-S1aNmsvIZaFFT6VFzU0f4-g,33
|
1256
|
+
tf_models_nightly-2.19.0.dev20250130.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|
File without changes
|