GitLab

PiperOrigin-RevId: 369375299

PiperOrigin-RevId: 368535232

This CL changes ResidualBlock and InvertedBottleneckBlock.

PiperOrigin-RevId: 368383954

For all modified calls to set_mixed_precision_policy(), the loss_scale argument was removed, as it cannot be passed if the nonexperimental API is used. For all such callers, the loss_scale is later used to explicitly create a LossScaleOptimizer, so removing the argument has no impact.

Switching to the non-experimental LossScaleOptimizer has no effect, as it has near identical behavior and all isinstance checks within the official models check for the non-experimental version.

PiperOrigin-RevId: 368101975

PiperOrigin-RevId: 367105004

This has no functional impact since the default is currently True, but I plan on changing the default to False soon.

PiperOrigin-RevId: 367049942

The function `tf.train.experimental.enable_mixed_precision_graph_rewrite` will be removed from the TF2 namespace soon, at which point it will only be accessible under tf.compat.v1.

PiperOrigin-RevId: 367046393

PiperOrigin-RevId: 365713370

PiperOrigin-RevId: 365085378

PiperOrigin-RevId: 363595377

PiperOrigin-RevId: 361348924

The default is True, but I plan on changing it to False soon. After that, I plan on removing the argument and never using the experimental API.

PiperOrigin-RevId: 360724698

PiperOrigin-RevId: 360719881

PiperOrigin-RevId: 360489888

PiperOrigin-RevId: 360256877

PiperOrigin-RevId: 359821383

[core] Use the full model instead of checkpoint_items when constructing Checkpoint in multitask evaluator.

PiperOrigin-RevId: 357962526

[core] Update BestCheckpointExporter to support dictionary with arbitrary depth (possibly coming from multitask eval).
Also added best checkpointexporter support to multitask.

PiperOrigin-RevId: 357584596

PiperOrigin-RevId: 356158181

PiperOrigin-RevId: 353327814

PiperOrigin-RevId: 353315689

PiperOrigin-RevId: 353277457

PiperOrigin-RevId: 353260280

PiperOrigin-RevId: 352912327

PiperOrigin-RevId: 352711257

PiperOrigin-RevId: 352676136

PiperOrigin-RevId: 351524712

PiperOrigin-RevId: 350705651

PiperOrigin-RevId: 350170448

PiperOrigin-RevId: 349861117

PiperOrigin-RevId: 349314516

PiperOrigin-RevId: 348291651

PiperOrigin-RevId: 348228865

PiperOrigin-RevId: 348113609

PiperOrigin-RevId: 347634117

PiperOrigin-RevId: 342925872

PiperOrigin-RevId: 342548143

PiperOrigin-RevId: 342354179