mttf 1.2.23__py3-none-any.whl → 1.2.24__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.

Potentially problematic release.


This version of mttf might be problematic. Click here for more details.

@@ -2,7 +2,7 @@ from .. import constraints
2
2
 
3
3
 
4
4
  class CenterAround(constraints.Constraint):
5
- """Constrains weight tensors to be centered around `ref_value`."""
5
+ """Constrains the last axis to have values centered around `ref_value`."""
6
6
 
7
7
  def __init__(self, ref_value: float = 0.0):
8
8
  self.ref_value = ref_value
@@ -10,8 +10,10 @@ class CenterAround(constraints.Constraint):
10
10
  def __call__(self, w):
11
11
  import tensorflow as tf
12
12
 
13
- mean = tf.reduce_mean(w)
14
- return w - (mean - self.ref_value)
13
+ mean = tf.reduce_mean(w, axis=-1, keepdims=True)
14
+ ref_mean = mean - self.ref_value
15
+ ref_mean = tf.expand_dims(ref_mean, -1)
16
+ return w - ref_mean
15
17
 
16
18
  def get_config(self):
17
19
  return {"ref_value": self.ref_value}
mt/tf/version.py CHANGED
@@ -1,5 +1,5 @@
1
1
  MAJOR_VERSION = 1
2
2
  MINOR_VERSION = 2
3
- PATCH_VERSION = 23
3
+ PATCH_VERSION = 24
4
4
  version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
5
5
  __all__ = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mttf
3
- Version: 1.2.23
3
+ Version: 1.2.24
4
4
  Summary: A package to detect and monkey-patch TensorFlow and Keras, for Minh-Tri Pham
5
5
  Home-page: https://github.com/inteplus/mttf
6
6
  Author: ['Minh-Tri Pham']
@@ -6,7 +6,7 @@ mt/keras_src/applications_src/classifier.py,sha256=rmJThMRoDRwROvcxRek5-NI6OSpGh
6
6
  mt/keras_src/applications_src/mobilenet_v3_split.py,sha256=EufzlbM2B5D_whpsDmfJFvyLSLtsoK6VXomaSE2K6Ms,19737
7
7
  mt/keras_src/applications_src/mobilevit.py,sha256=WSwTTT_VTPkH03XmRB_tFS1IgZf__HcVHwshjidSKaM,8934
8
8
  mt/keras_src/constraints_src/__init__.py,sha256=XOHrlZbWjmQszPm9TGQFSmKthbcgBvb1jXUlgZbtyMM,253
9
- mt/keras_src/constraints_src/center_around.py,sha256=8p7W9A9N7HdOjsdN05yC4ZTSLj9bUEeIvxWmC_Z61tE,437
9
+ mt/keras_src/constraints_src/center_around.py,sha256=NIPckqjbrOfGCT3Fv1O3hET4sHgE7kBWIMt5AG03Lmk,543
10
10
  mt/keras_src/layers_src/__init__.py,sha256=EnDoayl_SVb5cYyISiGbe0v5IPoklE4BgyJk-ft8SrI,746
11
11
  mt/keras_src/layers_src/counter.py,sha256=pFN50ZOpV4lvkKmKsK5hxfvAF0Jj46TtGqZOtND5eSc,852
12
12
  mt/keras_src/layers_src/floor.py,sha256=wPINhDnRYdbJ_eQqwq0pCXiez1sAryvGthQJULeN9yM,508
@@ -20,21 +20,21 @@ mt/tf/__init__.py,sha256=M8xiJNdrAUJZgiZTOQOdfkehjO-CYzGpoxh5HVGBkms,338
20
20
  mt/tf/init.py,sha256=bcm0t5tstxTkCBOiMX1SJxhKOz0MAZf-MYp2Mk0Gtas,502
21
21
  mt/tf/mttf_version.py,sha256=ha53i-H9pE-crufFttUECgXHwPvam07zMKzApUts1Gs,206
22
22
  mt/tf/utils.py,sha256=wau2vhPoPHu2cDxlc2lc9fxrndOXPdq2DNG4em5OOMI,1025
23
- mt/tf/version.py,sha256=Z9OUrBj7zV9Z3Ny_BZHgkvg8WQWYLBVpTdDVeYot2WM,207
23
+ mt/tf/version.py,sha256=6FaoVwSYTyaipLFHUyDW3EHsEl4mM_bdUbuwZOMtaHg,207
24
24
  mt/tf/keras_applications/__init__.py,sha256=m-A1rHGGLQgHX9690ENWXZkrU0vqfsJkZXcjIG3CLM0,142
25
25
  mt/tf/keras_layers/__init__.py,sha256=NsuFD-kSuy6cVV3Kl7ab95tw4g7x4Igv3cF-Ky3VuCo,124
26
26
  mt/tfc/__init__.py,sha256=pcxgcmUSXJCFZal5Pxz7hC1P8fhP1dB2egZzKGdrseQ,12118
27
27
  mt/tfg/__init__.py,sha256=6Ly2QImAyQTsg_ZszuAuK_L2n56v89Cix9yYmMVk0CM,304
28
28
  mt/tfp/__init__.py,sha256=AQkGCkmDRwswEt3qoOSpxe-fZekx78sHHBs2ZVz33gc,383
29
29
  mt/tfp/real_nvp.py,sha256=U9EmkXGqFcvtS2yeh5_RgbKlVKKlGFGklAb7Voyazz4,4440
30
- mttf-1.2.23.data/scripts/dmt_pipi.sh,sha256=NNsj4P332unHMqU4mAFjU9PQvxQ8TK5XQ42LC29IZY8,510
31
- mttf-1.2.23.data/scripts/dmt_twineu.sh,sha256=KZhcYwuCW0c36tWcOgCe7uxJmS08rz-J6YNY76Exy4M,193
32
- mttf-1.2.23.data/scripts/pipi.sh,sha256=kdo96bdaKq2QIa52Z4XFSiGPcbDm09SAU9cju6I2Lxo,289
33
- mttf-1.2.23.data/scripts/wml_nexus.py,sha256=47P9PQMgb9w_-T0olC-dr3s60mKaQup-RWOuNi5mvJg,1192
34
- mttf-1.2.23.data/scripts/wml_pipi.sh,sha256=CuidIcbuxyXSBNQqYRhCcSC8QbBaSGnQX0KAIFaIvKA,499
35
- mttf-1.2.23.data/scripts/wml_twineu.sh,sha256=av1JLN765oOWC5LPkv2eSWIVof26y60601tMGkuYdb8,180
36
- mttf-1.2.23.dist-info/licenses/LICENSE,sha256=e_JtcszdGZ2ZGfjcymTGrcxFj_9XPicZOVtnsrPvruk,1070
37
- mttf-1.2.23.dist-info/METADATA,sha256=IfuCD8V_lSwjPYzcFAfr30hbXZhG7eBt7_V1B8PNi_Q,568
38
- mttf-1.2.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
39
- mttf-1.2.23.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
40
- mttf-1.2.23.dist-info/RECORD,,
30
+ mttf-1.2.24.data/scripts/dmt_pipi.sh,sha256=NNsj4P332unHMqU4mAFjU9PQvxQ8TK5XQ42LC29IZY8,510
31
+ mttf-1.2.24.data/scripts/dmt_twineu.sh,sha256=KZhcYwuCW0c36tWcOgCe7uxJmS08rz-J6YNY76Exy4M,193
32
+ mttf-1.2.24.data/scripts/pipi.sh,sha256=kdo96bdaKq2QIa52Z4XFSiGPcbDm09SAU9cju6I2Lxo,289
33
+ mttf-1.2.24.data/scripts/wml_nexus.py,sha256=47P9PQMgb9w_-T0olC-dr3s60mKaQup-RWOuNi5mvJg,1192
34
+ mttf-1.2.24.data/scripts/wml_pipi.sh,sha256=CuidIcbuxyXSBNQqYRhCcSC8QbBaSGnQX0KAIFaIvKA,499
35
+ mttf-1.2.24.data/scripts/wml_twineu.sh,sha256=av1JLN765oOWC5LPkv2eSWIVof26y60601tMGkuYdb8,180
36
+ mttf-1.2.24.dist-info/licenses/LICENSE,sha256=e_JtcszdGZ2ZGfjcymTGrcxFj_9XPicZOVtnsrPvruk,1070
37
+ mttf-1.2.24.dist-info/METADATA,sha256=G0n_8yIDPY33sfF41lqDz7LjBqmxGRRR0Er7YN4rl2M,568
38
+ mttf-1.2.24.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
39
+ mttf-1.2.24.dist-info/top_level.txt,sha256=WcqGFu9cV7iMZg09iam8eNxUvGpLSKKF2Iubf6SJVOo,3
40
+ mttf-1.2.24.dist-info/RECORD,,
File without changes
File without changes