Thanks to visit codestin.com
Credit goes to github.com

Commits

Commits on Sep 26, 2022

Update the V1 Eager checkpointer path so it's compatible with the Graph mode trainer

lingvo-bot
authored and
copybara-github
committed

Commits on Sep 24, 2022

Silence new type errors in Python 3.10.

lingvo-bot
authored and
copybara-github
committed
Cleanup: avoid referring to `std::vector<T>` members when `T` is incomplete.

lingvo-bot
authored and
copybara-github
committed

Commits on Sep 22, 2022

Define array without concatenation.

lingvo-bot
authored and
copybara-github
committed
Fix typo when decoder metrics pass_through_transcript_field=true and only_output_tpu_tensors=true, delete key 'transcripts'/'topk_decoded' only when they exist.

lingvo-bot
authored and
copybara-github
committed
Add _MaybeCheckParamsDuringInitAfterTrialOverride to BaseRunner.

lingvo-bot
authored and
copybara-github
committed

Commits on Sep 21, 2022

Add a flag to support the Trainer class driving checkpointing

freeblee
authored and
copybara-github
committed
1. Added TF2 mode test trainer_tpu_eager_test (skipped some of them)

lingvo-bot
authored and
copybara-github
committed

Commits on Sep 19, 2022

The main purpose of the CL is to add TF2 tests for executor_tpu_models_test.py. I also included code changes necessary for executor_tpu_models_eager_test:

lingvo-bot
authored and
copybara-github
committed

Commits on Sep 15, 2022

Replace `tensorflow::Status::OK()` with` tensorflow::OkStatus()`.

lingvo-bot
authored and
copybara-github
committed

Commits on Sep 14, 2022

1. Exposed flag `SetReplaceAliasInplaceUpdateInAttention` so we can replace `alias_inplace_update` with `tensor_scatter_nd_update` to run tests on CPU

lingvo-bot
authored and
copybara-github
committed

Commits on Sep 13, 2022

Commits on Sep 12, 2022

Allow infering TPU cluster parameters in internal clusters.

lingvo-bot
authored and
copybara-github
committed

Commits on Sep 9, 2022

xla sharding flag is set depending on device_mesh param.

lingvo-bot
authored and
copybara-github
committed
Make cyclical cosine schedule restart every pi.

ds-hwang
authored and
copybara-github
committed

Commits on Sep 8, 2022

Commits on Sep 7, 2022

Commits on Sep 6, 2022

Add a flag to enable experimental async checkpointing when writing V2 checkpoints in eager.

lingvo-bot
authored and
copybara-github
committed

Commits on Sep 2, 2022

Add options to skip checking that all variables are consumed in `RestoreFromPath()`.

lingvo-bot
authored and
copybara-github
committed
Enforce StackingOverTime's input to have rank three.

lingvo-bot
authored and
copybara-github
committed
Enable weight only quantization on gshard.

lingvo-bot
authored and
copybara-github
committed

Commits on Aug 31, 2022

Remove unnecessary ToStaticShape calls.

lingvo-bot
authored and
copybara-github
committed
Change log message from INFO to VLOG(0).

lingvo-bot
authored and
copybara-github
committed
Fix output dimension of matmuls in ProjectionLayer: should be self._internal_output_dim, or else the layer will fail with *GLU activations.

lingvo-bot
authored and
copybara-github
committed

Commits on Aug 29, 2022

Commits on Aug 25, 2022