c92ac8e | Chen Qian | 16 August 2022, 19:22:00 UTC | Use legacy keras optimizer to be compatible with an incoming Keras optimizer migration. PiperOrigin-RevId: 467992914 | 16 August 2022, 19:22:29 UTC |
d1cd371 | Yilei Yang | 07 March 2022, 21:15:40 UTC | Update test for Python 3.9. PiperOrigin-RevId: 433021769 | 07 March 2022, 21:16:14 UTC |
df5d099 | Sonnet Contributor | 01 February 2022, 18:27:44 UTC | Replace np.bool with np.bool_. As of NumPy 1.20, `np.bool` is an alias of builtin type `bool`. The numpy scalar type is `np.bool_`. PiperOrigin-RevId: 425658873 | 01 February 2022, 18:28:37 UTC |
5cbfdc3 | Tom Hennigan | 15 December 2021, 15:35:49 UTC | Defer creation of autograph converted functions until usage. Some users (e.g. #226) report issues with autograph. While these can't be fixed by Sonnet, we can defer the creation of autograph converted functions until they are first used (e.g. until the first time you use batch norm with `@tf.function`). This enables users to make use of non-autograph requiring features in Sonnet and may be a sufficient workaround while TF team debug. Fixes #226. PiperOrigin-RevId: 416554741 | 15 December 2021, 15:36:30 UTC |
958ea1b | Tom Hennigan | 13 December 2021, 12:51:03 UTC | Bump Sonnet version to `2.0.1.dev`. PiperOrigin-RevId: 416012688 | 13 December 2021, 12:51:34 UTC |
913cbd3 | Mehdi Amini | 30 November 2021, 10:17:47 UTC | Disable flaky test from CI PiperOrigin-RevId: 413094278 | 30 November 2021, 10:18:21 UTC |
19e238e | Mehdi Amini | 17 October 2021, 14:26:53 UTC | Tag flaky test sonnet/v2/src/conformance:checkpoint_test_gpu as "notap" PiperOrigin-RevId: 403759548 | 17 October 2021, 14:27:22 UTC |
892790c | Copybara-Service | 08 September 2021, 08:52:39 UTC | Merge pull request #214 from deepmind:dependabot/pip/tensorflow-2.5.1 PiperOrigin-RevId: 395424166 | 08 September 2021, 08:52:39 UTC |
0b550ff | Sergei Lebedev | 02 September 2021, 10:03:58 UTC | Added __init__.py files to all src subpackages PiperOrigin-RevId: 394419068 | 02 September 2021, 10:04:40 UTC |
57f4462 | dependabot[bot] | 25 August 2021, 14:45:08 UTC | Bump tensorflow from 2.5.0 to 2.5.1 Bumps [tensorflow](https://github.com/tensorflow/tensorflow) from 2.5.0 to 2.5.1. - [Release notes](https://github.com/tensorflow/tensorflow/releases) - [Changelog](https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md) - [Commits](https://github.com/tensorflow/tensorflow/compare/v2.5.0...v2.5.1) --- updated-dependencies: - dependency-name: tensorflow dependency-type: direct:production ... Signed-off-by: dependabot[bot] <support@github.com> | 25 August 2021, 14:45:08 UTC |
0e25f47 | Sonnet Contributor | 13 August 2021, 18:11:51 UTC | N/A BUILD file cleanup PiperOrigin-RevId: 390652227 | 13 August 2021, 18:12:21 UTC |
38adf3a | Tom Hennigan | 12 August 2021, 15:56:21 UTC | Mark embeddings on VectorQuantizerEMA as non-trainable. This variable is updated as part of the forward pass (during training) and is not updated by the optimiser. In TF `trainable` is used to indicate variables that should be managed by an optimizer, so even though we update this during training and it is a parameter of our model, we should set `trainable=False`. Fixes #209. PiperOrigin-RevId: 390382636 | 12 August 2021, 15:56:52 UTC |
c87468e | Sergei Lebedev | 18 June 2021, 19:01:55 UTC | Removed six Sonnet 2 does not support 2.X, so it makes little sense to keep six around. PiperOrigin-RevId: 380235488 | 18 June 2021, 19:02:30 UTC |
cf15370 | Tom Hennigan | 18 June 2021, 12:46:27 UTC | Remove Python 2 supporting code. PiperOrigin-RevId: 380168270 | 18 June 2021, 12:46:56 UTC |
9bb751f | Tom Hennigan | 18 June 2021, 09:55:21 UTC | Add functional API to Sonnet 2 inspired by JAX [0] and Haiku [1]. [0] https://github.com/google/jax [1] https://github.com/deepmind/dm-haiku PiperOrigin-RevId: 380147873 | 18 June 2021, 09:55:58 UTC |
6c2bf5f | Tom Hennigan | 10 June 2021, 10:05:52 UTC | Add pytest github action and get tests passing. PiperOrigin-RevId: 378611382 | 10 June 2021, 10:06:20 UTC |
1a5a6c4 | Tom Hennigan | 09 June 2021, 11:24:02 UTC | Relax result checking for FP values. CUDA-11 produces slightly different results and causes the test failure otherwise. PiperOrigin-RevId: 378374202 | 09 June 2021, 11:24:48 UTC |
81baad2 | Rebecca Chen | 01 June 2021, 18:28:39 UTC | Add missing typing.Optional type annotations to function parameters. PiperOrigin-RevId: 376879768 Change-Id: I6efe84d5f989d8d5663d7ab6b5b3559fe071ade3 | 01 June 2021, 18:29:10 UTC |
c9cdbb4 | Eugene Zhulenev | 15 March 2021, 18:22:54 UTC | Fix sonnet/v2/src/nets/dnc:util_test_gpu test PiperOrigin-RevId: 362986883 Change-Id: If791985603e97af2f0bca8c2255a2ba995f27b28 | 15 March 2021, 18:23:31 UTC |
01ef9a6 | Yilei Yang | 11 March 2021, 15:29:15 UTC | Remove no-op pylint disable comments. PiperOrigin-RevId: 362287115 Change-Id: Ib5efc634f54914055d10b914f7aac70d386edf7e | 11 March 2021, 15:29:48 UTC |
76f61d5 | Tamara Norman | 11 March 2021, 14:35:47 UTC | Remove tensorflow-gpu from extra requirements Also update dependencies in requirements-tf.txt PiperOrigin-RevId: 362278786 Change-Id: Ib8943e04c36a7fe31782aecaede8cd98bfd69347 | 11 March 2021, 14:36:11 UTC |
eb06599 | Isha Arkatkar | 10 February 2021, 05:14:47 UTC | Fix the checkpoint_test after fix for bug b/168905859 PiperOrigin-RevId: 356659488 Change-Id: Ibd52064ff9c343cf10eee5bb2720e04e10ccb421 | 10 February 2021, 05:15:36 UTC |
5a46569 | Copybara-Service | 08 October 2020, 10:01:11 UTC | Merge pull request #181 from tirkarthi:fix-collections PiperOrigin-RevId: 336048465 Change-Id: I93e579d4e9832aa9ca66a68fac04ea88183970ee | 08 October 2020, 10:01:11 UTC |
af46202 | Copybara-Service | 01 October 2020, 13:24:02 UTC | Merge pull request #190 from deepmind:dependabot/pip/tensorflow-2.2.1 PiperOrigin-RevId: 334799980 Change-Id: Ie453fa1ca5cf42d06c5d1bfdd952952cee518011 | 01 October 2020, 13:24:02 UTC |
c3e07d3 | Sonnet Contributor | 30 September 2020, 22:58:35 UTC | BUILD cleanup PiperOrigin-RevId: 334697535 Change-Id: I738c37f5b79f02bd9ab5d0c027858293ffbeb58a | 01 October 2020, 13:23:41 UTC |
91a4c09 | Lena Martens | 14 September 2020, 13:55:50 UTC | Internal change. PiperOrigin-RevId: 331537710 Change-Id: Ie0374ee84cc781870ad0717b59a6aaacbfa3fc44 | 01 October 2020, 13:23:23 UTC |
af3826a | dependabot[bot] | 01 October 2020, 10:15:16 UTC | Bump tensorflow from 2.2.0 to 2.2.1 Bumps [tensorflow](https://github.com/tensorflow/tensorflow) from 2.2.0 to 2.2.1. - [Release notes](https://github.com/tensorflow/tensorflow/releases) - [Changelog](https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md) - [Commits](https://github.com/tensorflow/tensorflow/compare/v2.2.0...v2.2.1) Signed-off-by: dependabot[bot] <support@github.com> | 01 October 2020, 10:15:16 UTC |
4d16e10 | Sonnet Contributor | 09 September 2020, 12:35:53 UTC | Internal refactor PiperOrigin-RevId: 330704150 Change-Id: Ib372f470428f41b2656a6d680d86bca73b973a30 | 01 October 2020, 09:20:16 UTC |
3241e36 | Sonnet Contributor | 07 September 2020, 18:51:40 UTC | Internal change PiperOrigin-RevId: 330403382 Change-Id: Ie0ea28603cef70e844cc291f9a4e55fc9d4f9be4 | 07 September 2020, 18:52:09 UTC |
77792eb | Malcolm Reynolds | 07 September 2020, 18:38:15 UTC | Internal change PiperOrigin-RevId: 330402541 Change-Id: I37a1c7a3cf9f42635138c57ad0eb26dce3d580d3 | 07 September 2020, 18:38:46 UTC |
2827d95 | Adrian Kuegel | 24 July 2020, 07:21:12 UTC | Change saved_model_test back to use assertAllClose. We are dealing here with floating point values, so some small errors can happen. PiperOrigin-RevId: 322948216 Change-Id: I4dc8e7f728f6e3500bbf5546bbba2b418d14b13f | 24 July 2020, 07:21:44 UTC |
74be631 | Tom Hennigan | 16 July 2020, 11:11:35 UTC | Ensure data dependence on input to saved model. PiperOrigin-RevId: 321539026 Change-Id: If585d4dfeb8d481e966c39721ef41e4cba37799c | 16 July 2020, 11:12:04 UTC |
45b2b15 | Karthikeyan Singaravelan | 09 July 2020, 14:36:59 UTC | Import ABC from collections.abc | 09 July 2020, 14:36:59 UTC |
be8e9eb | Tom Hennigan | 02 July 2020, 08:50:10 UTC | Rename whitelist to allow. PiperOrigin-RevId: 319373568 Change-Id: If0a500774fef2a67e7bef4ed8b63f976c8b038e2 | 02 July 2020, 08:50:39 UTC |
089e3b0 | Ruoxin Sang | 01 July 2020, 17:50:11 UTC | Update TPUStrategy symbol in Sonnet. PiperOrigin-RevId: 319255469 Change-Id: I5cf0d3596cc5e2ffffc4b93ae8184fa2d3f43dd6 | 01 July 2020, 17:50:46 UTC |
ae34a78 | Tom Hennigan | 14 May 2020, 13:01:33 UTC | Use tfds for Cifar10 in VQ-VAE example, default to EMA and include cell outputs. PiperOrigin-RevId: 311517163 Change-Id: Ie8ebc0f95c14d6e70118f48695be65ad7773a13b | 14 May 2020, 13:02:05 UTC |
59f8a46 | Tom Hennigan | 09 May 2020, 11:52:43 UTC | Run tests with TensorFlow 2.2.0. PiperOrigin-RevId: 310708598 Change-Id: I7f4b141b4354931d6d241762b9419eaed1b46d46 | 09 May 2020, 11:53:07 UTC |
61b97be | Sonnet Contributor | 04 May 2020, 15:17:28 UTC | Change BatchApply to accept any callable that returns a tf.Tensor. PiperOrigin-RevId: 309746386 Change-Id: I6e7331e9e7facd58f8f5e87e1ba6037bd26e0c2c | 04 May 2020, 15:17:57 UTC |
550d8c6 | Tom Hennigan | 24 April 2020, 20:00:57 UTC | Use sphinx_autodoc_typehints for python types. PiperOrigin-RevId: 308309638 Change-Id: Ie8d2adb67ba0d3159dda3ca255b0b02c0a36d64b | 24 April 2020, 20:01:37 UTC |
5c5ef16 | Tom Hennigan | 20 April 2020, 13:48:57 UTC | Drop usages of `tf.enable_v2_behavior()`, no longer necessary. PiperOrigin-RevId: 307389325 Change-Id: Id5d59c4303dd6c10c0328f801051e95de1bdab23 | 20 April 2020, 13:49:25 UTC |
4633390 | Copybara-Service | 17 April 2020, 18:23:15 UTC | Merge pull request #168 from ialong:patch-1 PiperOrigin-RevId: 307079951 Change-Id: Ic91440e5fe76aec3cf46d39212b4dd4155a42589 | 17 April 2020, 18:23:15 UTC |
a06702c | Tom Hennigan | 17 April 2020, 12:53:37 UTC | Update distributed training notes. PiperOrigin-RevId: 307028907 Change-Id: I5697668bf11f0037d0b580b79973f6904bad4222 | 17 April 2020, 12:54:05 UTC |
baf0c57 | Tom Hennigan | 17 April 2020, 08:19:09 UTC | Use latest version of Sonnet in colab examples. Fixes #169. PiperOrigin-RevId: 307003064 Change-Id: I17ac33a29c2dd88ff38d2933adf80c04b2fd27c5 | 17 April 2020, 08:19:41 UTC |
018b05e | Alessandro Davide Ialongo | 15 April 2020, 16:46:32 UTC | add float64 test for VQ-VAE make sure `VectorQuantizerEMA` works if the dtype is tf.float64 | 15 April 2020, 16:46:32 UTC |
58d9a27 | Alessandro Davide Ialongo | 12 April 2020, 12:31:41 UTC | fix type of one hot indicators in vqvae.py `tf.one_hot` defaults to `tf.float32` if no dtype is provided. This prevents `VectorQuantizerEMA` from running if its dtype is specified as `tf.float64` at init time (since the EMA tries to subtract a `tf.float32` - the one hot indicator tensor - from a `tf.float64`). This is easily fixed by passing a dtype argument to `tf.one_hot`. | 12 April 2020, 12:31:41 UTC |
941f7b3 | Mark Daoust | 03 April 2020, 18:07:54 UTC | Remove `%tensorflow_version` and `from __future__` PiperOrigin-RevId: 304652978 Change-Id: I4e46902b49b71a98cd3ebc13fd5eb20e4c076dbf | 03 April 2020, 18:08:20 UTC |
3b8f9d4 | Tamara Norman | 02 April 2020, 14:13:06 UTC | Fix for change to conv_transpose PiperOrigin-RevId: 304394090 Change-Id: I7cdbfde46fc70a7c324ac8d7b4fc50ef89516639 | 02 April 2020, 14:13:31 UTC |
f28eb90 | Tamara Norman | 02 April 2020, 12:48:58 UTC | Support shape changing inputs PiperOrigin-RevId: 304382117 Change-Id: I8b2f86adc492365c3327770aa69db01c62c1aebc | 02 April 2020, 12:49:27 UTC |
67a0955 | John Aslanides | 27 March 2020, 11:08:06 UTC | Update README to reflect 2.0 release. PiperOrigin-RevId: 303301305 Change-Id: Ifdef58d5cf7170da8b15bccaea7a2b8d692ec1ae | 27 March 2020, 11:08:37 UTC |
940b4d1 | Tom Hennigan | 27 March 2020, 10:05:43 UTC | Update Sonnet version to 2.0.0. PiperOrigin-RevId: 303292840 Change-Id: I0118990892ac9c369c94de966a092f5056d1eafd | 27 March 2020, 10:06:14 UTC |
bb53c63 | Ken Franko | 19 March 2020, 23:32:05 UTC | Apply name change(experimental_run_v2 -> run) for all callers. PiperOrigin-RevId: 301919882 Change-Id: I14c6ed85bdf50d619d1bc572e0fbcc5f1821c70b | 19 March 2020, 23:32:32 UTC |
de30fda | Tamara Norman | 16 March 2020, 12:28:44 UTC | Ensure batch norm works for inference when the number of channels is 1 PiperOrigin-RevId: 301137212 Change-Id: I8c9d03dce81d7b38b2d082873c9fba75662541b3 | 16 March 2020, 12:29:09 UTC |
7a70496 | Sonnet Copybara | 09 March 2020, 14:43:46 UTC | Merge pull request #164 from hanbyul-kim:hotfix/resolve_doc_warnings PiperOrigin-RevId: 299838560 Change-Id: I8698eed36a2b122fdfc11a83a92925da8ae5a2fb | 09 March 2020, 14:43:46 UTC |
733d6bb | Tamara Norman | 27 February 2020, 14:56:39 UTC | Fix typing of cifar10_convnet after previous change PiperOrigin-RevId: 297585986 Change-Id: Icdcfbdec09feafb1026385bf794d252a626a8219 | 27 February 2020, 14:57:06 UTC |
3debabb | Tamara Norman | 27 February 2020, 14:12:47 UTC | Internal change PiperOrigin-RevId: 297579572 Change-Id: Id57c06efd031bae6449ae02f4eb361e5ddacde38 | 27 February 2020, 14:13:15 UTC |
2a8e1d8 | Sonnet Contributor | 26 February 2020, 19:45:26 UTC | Explicitly create names for batch_norm moving statistics. PiperOrigin-RevId: 297411469 Change-Id: I6c9c37958930614bf47f10edf51dafb161b2ace3 | 26 February 2020, 19:46:05 UTC |
c8fbd10 | Hanbyul Kim | 26 February 2020, 14:15:04 UTC | Fix indentation error | 26 February 2020, 14:15:04 UTC |
8574df2 | Hanbyul Kim | 26 February 2020, 13:35:21 UTC | Split a line into two and remove indent | 26 February 2020, 13:39:04 UTC |
e7f7b61 | Hanbyul Kim | 26 February 2020, 13:34:00 UTC | Remove extra newlines | 26 February 2020, 13:39:04 UTC |
623a88d | Hanbyul Kim | 26 February 2020, 13:19:04 UTC | Fix code block to use :: | 26 February 2020, 13:39:04 UTC |
2095e03 | Hanbyul Kim | 25 February 2020, 14:46:58 UTC | Fix broken code block | 26 February 2020, 13:39:04 UTC |
dab662d | Hanbyul Kim | 25 February 2020, 14:31:31 UTC | Fix unexpected indent of docstring | 26 February 2020, 13:39:04 UTC |
d0cc578 | Sonnet Contributor | 26 February 2020, 12:32:43 UTC | In preparation for cl/296449699, disabling some pytype checks that fail after this CL is submitted. The root cause for these failures is that currently tensorflow.compat.v1 maps to Any and effectively there isn't any type checking. cl/296449699 enables this type checking for tensorflow.compat.v1 and as a result exposes some bugs / problems with the type checking code that got checked in previously. PiperOrigin-RevId: 297331371 Change-Id: I7bb4ccb53ef6aebcae69e809ce245b1f02882ea9 | 26 February 2020, 12:33:18 UTC |
3732f19 | Sonnet Contributor | 24 February 2020, 13:30:51 UTC | DepthwiseConv2D is now accessible via snt.DepthwiseConv2D PiperOrigin-RevId: 296870777 Change-Id: Ie1ac45cfbb8cee5b004ddffe34568044405245dc | 24 February 2020, 13:31:20 UTC |
1f5c0c2 | Sonnet Copybara | 21 February 2020, 09:25:53 UTC | Merge pull request #159 from chris-chris:feature/2002-mac-test PiperOrigin-RevId: 296389674 Change-Id: Ifce8e017fbbce4617657f958e83667e8d71e51bd | 21 February 2020, 09:25:53 UTC |
d4f38d4 | Sergei Lebedev | 20 February 2020, 20:09:08 UTC | Fixed a few formatting quirks in recurrent docs PiperOrigin-RevId: 296267592 Change-Id: I9094ae88cf8ef16ee3721c8bd8f19532008a0f13 | 20 February 2020, 20:09:55 UTC |
4cbbe29 | Hoyean Song | 20 February 2020, 19:37:05 UTC | CPU count on macos | 20 February 2020, 19:37:05 UTC |
12c3b22 | Ken Franko | 19 February 2020, 18:05:28 UTC | Remove values property from DistributedValues. PiperOrigin-RevId: 295994651 Change-Id: I663f2b2b94a04e4a9ac01ca63629d7bbf412ea15 | 19 February 2020, 18:05:49 UTC |
b6152b7 | Ken Franko | 18 February 2020, 22:56:03 UTC | Make primary property on DistributedValue private. PiperOrigin-RevId: 295829087 Change-Id: I5c0bf035780f92de197578c979301ac900dc7afc | 18 February 2020, 22:56:38 UTC |
4bbbcbc | Mehdi S. M. Sajjadi | 10 February 2020, 12:28:06 UTC | Make snt.Linear compatible with more than one batch dimension. The Linear layer should be compatible with inputs of dimension 2 or more. All dimensions but the last one will be treated as batch dimensions. PiperOrigin-RevId: 294193315 Change-Id: Ia673936f09832dabfaea02e612f34a59c8da187e | 10 February 2020, 12:28:28 UTC |
f9cc3e9 | Sonnet Copybara | 30 January 2020, 14:18:49 UTC | Merge pull request #157 from deepmind:dependabot/pip/tensorflow-2.0.1 PiperOrigin-RevId: 292334684 Change-Id: Ib2fff7befb0c83110abe292dcdfbe144a549f41b | 30 January 2020, 14:18:49 UTC |
9eaac57 | dependabot[bot] | 28 January 2020, 21:46:22 UTC | Bump tensorflow from 2.0.0 to 2.0.1 Bumps [tensorflow](https://github.com/tensorflow/tensorflow) from 2.0.0 to 2.0.1. - [Release notes](https://github.com/tensorflow/tensorflow/releases) - [Changelog](https://github.com/tensorflow/tensorflow/blob/master/RELEASE.md) - [Commits](https://github.com/tensorflow/tensorflow/compare/v2.0.0...v2.0.1) Signed-off-by: dependabot[bot] <support@github.com> | 28 January 2020, 21:46:22 UTC |
f18c55e | Tom Hennigan | 27 January 2020, 14:23:29 UTC | Fuzz tests for more optimizers. PiperOrigin-RevId: 291709836 Change-Id: Ia1282ba02649863e1ea39c481a84e31f68ace7ea | 27 January 2020, 14:23:57 UTC |
4fe2373 | Sonnet Copybara | 27 January 2020, 09:46:12 UTC | Merge pull request #155 from fostiropoulos:v2 PiperOrigin-RevId: 291678292 Change-Id: I390276d492db8e7b1aaaf994b9e8592391d2e7ad | 27 January 2020, 09:46:12 UTC |
82049c2 | fostiropoulos | 25 January 2020, 19:48:24 UTC | Fixed vqvae_example.ipynb to be compatible with tf v2 | 25 January 2020, 19:48:24 UTC |
c616bbd | Tom Hennigan | 22 January 2020, 20:12:17 UTC | Expose `snt.{merge,split}_leading_dims`. PiperOrigin-RevId: 291003968 Change-Id: I7678d4aa15bb08650203afe0e21da9141c96a574 | 22 January 2020, 20:12:43 UTC |
af606a4 | Tom Hennigan | 22 January 2020, 09:52:41 UTC | Shard conformance tests to allow faster testing. PiperOrigin-RevId: 290911624 Change-Id: Iacf31a4798c428942708acc43ecf0d9f071ce88c | 22 January 2020, 09:53:17 UTC |
f44c133 | Tom Hennigan | 19 January 2020, 13:16:26 UTC | Create `snt.distribute.create_variables_eagerly(f)`. `create_variables_eagerly` is a higher order function that attempts to lift all variable creation within a given function into eager mode. It does this by patching Sonnet initializers to run under a `tf.init_scope`, and then patching variable creation to also run under a `tf.init_scope`. Additionally we support statically determining some values not created using Sonnet (e.g. `tf.zeros`). This function is useful when doing large scale distributed training with TensorFlow on clusters of machines. We can avoid much of the checking that happens for variables created in tf.functions which result in a lot of RPCs between the coordinator and workers, slowing down the first training step. PiperOrigin-RevId: 290499884 Change-Id: I1fce83011df198ff23c044689b6e5d0e60094be2 | 19 January 2020, 13:16:53 UTC |
25f040a | Sonnet Contributor | 17 January 2020, 10:33:27 UTC | Fix bug in _rnn_step, by swapping the prev_state with state. PiperOrigin-RevId: 290235436 Change-Id: Iea5fb1d0b6a14e6f050212293861b9fc107b7e9c | 17 January 2020, 10:34:00 UTC |
d8d1652 | Chris Jones | 15 January 2020, 14:40:14 UTC | Don't run stateful mixed precision example in doctest. PiperOrigin-RevId: 289848542 Change-Id: I66743dd05dccdac8a8b063b73995601a6c430652 | 15 January 2020, 14:40:38 UTC |
0c623c0 | Tom Hennigan | 08 January 2020, 13:04:39 UTC | Expose block group and bottle neck blocks from resnet. PiperOrigin-RevId: 288675258 Change-Id: Id3eb4de377f09a497fb5cfd680d7f898e4413e3a | 08 January 2020, 13:05:11 UTC |
431c6e4 | Tamara Norman | 07 January 2020, 15:20:02 UTC | Add check on inputs and outputs for TPU strategy test PiperOrigin-RevId: 288491883 Change-Id: I6f1b8db69ca360d152bf6eeb21bf1c06aeedbf92 | 07 January 2020, 15:20:57 UTC |
6e5de35 | Ken Franko | 06 January 2020, 16:54:31 UTC | Add cross-replica BatchNorm test on TPU strategy. PiperOrigin-RevId: 288309269 Change-Id: Ibe831d68d273fd9d279e663625d1fed7e84bba46 | 06 January 2020, 16:55:03 UTC |
1d16fc1 | Sonnet Contributor | 02 January 2020, 18:20:12 UTC | Remove eager variable creation hack. Underlying performance issue has been fixed. Workaround no longer needed. PiperOrigin-RevId: 287854778 Change-Id: Idc9609bf73baa432700bc2fe69dafefdafd970c6 | 02 January 2020, 18:20:45 UTC |
361f033 | Tom Hennigan | 02 January 2020, 09:05:41 UTC | Enable ResNet50 V2, has reproduced canonical experiment. PiperOrigin-RevId: 287803490 Change-Id: Id59f82145be188d1f587602d604c5aa12e9f5e2d | 02 January 2020, 09:06:13 UTC |
d7f1998 | Sergei Lebedev | 18 December 2019, 10:34:29 UTC | Removed wrapt.ObjectProxy hack from TrainableState tree now supports wrapped objects in the same way as tf.nest PiperOrigin-RevId: 286154954 Change-Id: I35dfeedf96be5d85bb1b74e2d79961e91172175b | 18 December 2019, 10:34:57 UTC |
6c6c533 | Sergei Lebedev | 16 December 2019, 13:39:05 UTC | Migrated from tf.nest to tree PiperOrigin-RevId: 285753613 Change-Id: I444db9927b81e04fb46a55edf44705fd2d511aad | 16 December 2019, 13:39:29 UTC |
84817e5 | Sonnet Contributor | 10 December 2019, 12:23:03 UTC | Fix or ignore type errors generated by the next release of pytype. The next pytype release includes better support for quoted annotations, which reveals type errors that were previously hidden due to pytype occasionally treating such annotations as Any rather than the quoted type. PiperOrigin-RevId: 284742182 Change-Id: Iba35edc283680295da815ee3c74a48eb67d158a0 | 10 December 2019, 12:23:29 UTC |
1928424 | Sonnet Contributor | 06 December 2019, 21:57:57 UTC | Throw an explicit error if user call TPUStrategy experimental_run_v2 in eager mode with a python function. PiperOrigin-RevId: 284256438 Change-Id: Iad99f48dab9787d1a53fb3e4815a357dbf147cdc | 06 December 2019, 21:58:24 UTC |
2ef5c4f | Tom Hennigan | 04 December 2019, 16:48:01 UTC | Drop bayes_by_backprop.py and remove dependency on tfp. PiperOrigin-RevId: 283764831 Change-Id: If8704c77238c4cc2b3650d58559d7ff833d44536 | 04 December 2019, 16:48:28 UTC |
50413fe | Sonnet Contributor | 15 November 2019, 14:45:02 UTC | Minor fixes. PiperOrigin-RevId: 280648498 Change-Id: I117af823bf44b3c58c20639008656d128a75891e | 15 November 2019, 14:45:30 UTC |
0fcfe87 | Tom Hennigan | 06 November 2019, 15:01:27 UTC | Ensure all builtin modules create and use parameters in a consistent order. Currently our optimizers rely on parameters being presented in a consistent order between calls to `opt.apply(..)`. This is because we use lists (because of checkpointing) to store momentum terms in Adam/Momentum etc. PiperOrigin-RevId: 278849811 Change-Id: I442681c9adcbcd08abce4a41fec3e1a8dd5a91a3 | 06 November 2019, 15:01:56 UTC |
d8557c3 | Malcolm Reynolds | 06 November 2019, 11:45:07 UTC | VQVAE example notebook ported to Sonnet 2. PiperOrigin-RevId: 278825348 Change-Id: I341cb0ee0a01d0ec93ba477a4489bc9ef3883969 | 06 November 2019, 11:45:35 UTC |
f9cbc4f | Tom Hennigan | 04 November 2019, 18:45:33 UTC | Re-add eager variable creation hack for TPUs. PiperOrigin-RevId: 278415088 Change-Id: I8cede4c69ebfd37fb606614647dccba1e7de49b9 | 04 November 2019, 18:46:06 UTC |
605ca70 | Malcolm Reynolds | 04 November 2019, 13:02:06 UTC | Fix bug in Conv*Transpose weight initialization PiperOrigin-RevId: 278353842 Change-Id: I8321c3488efd49304c3efe6a05e319217a80903f | 04 November 2019, 13:02:39 UTC |
43908bb | Tom Hennigan | 03 November 2019, 22:18:49 UTC | Move code for snt.distribute into src/distribute. PiperOrigin-RevId: 278255130 Change-Id: Ieeb64c6089c0c8ffc62135f991d8418c10514f4c | 03 November 2019, 22:19:23 UTC |
e3ad61e | Tom Hennigan | 01 November 2019, 16:33:35 UTC | Align names and defaults with Adam paper, add reference. PiperOrigin-RevId: 277942532 Change-Id: Ie660aae26b25a74ff55f06bf7bfb12790dcde80a | 01 November 2019, 16:34:11 UTC |
c787bec | Tom Hennigan | 30 October 2019, 17:50:12 UTC | Remove eager variable creation logic and tests from `replicator.py`. PiperOrigin-RevId: 277538633 Change-Id: I92b4568094893b7ee50a5765218f629de7fac700 | 30 October 2019, 17:50:39 UTC |
790bd15 | Sonnet Contributor | 30 October 2019, 16:39:59 UTC | Support restoring on creation for Sonnet model with TPUReplicator. PiperOrigin-RevId: 277523053 Change-Id: I2507f5ea24214c7e7465bb2ca6fb692758321382 | 30 October 2019, 16:40:28 UTC |
61cd600 | Chris Jones | 30 October 2019, 10:59:07 UTC | Make `BayesByBackprop` a Sonnet module. PiperOrigin-RevId: 277475497 Change-Id: Iafe3469bbb6a1ed2a28a4365491306d96b9c1d56 | 30 October 2019, 10:59:36 UTC |