GitLab

PiperOrigin-RevId: 357758634

PiperOrigin-RevId: 357746597

[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: 357284280

Update the other two colabs customize_encoder.ipynb and nlp_modeling_library_intro.ipynb to use the latest tensorflow models pip package.

PiperOrigin-RevId: 357275077

Update fine_tuning_bert.ipynb colab to use tf-models-official==2.4.0 and use the new checkpoint and tfhub model.

PiperOrigin-RevId: 357219857

PiperOrigin-RevId: 357209234

PiperOrigin-RevId: 357078424

PiperOrigin-RevId: 356907177

PiperOrigin-RevId: 356906879

PiperOrigin-RevId: 356906020

PiperOrigin-RevId: 356892700

PiperOrigin-RevId: 356629859

PiperOrigin-RevId: 356587847

PiperOrigin-RevId: 356580296

PiperOrigin-RevId: 356578560

PiperOrigin-RevId: 356306227

PiperOrigin-RevId: 356197222

PiperOrigin-RevId: 356158181

- Create shadow weights at the beginning of training.
- Swap the weights during.
- Save best checkpoint with average weights.

The following fields need to be set in order to activate the best checkpoint exporter.
best_checkpoint_eval_metric
best_checkpoint_export_subdir
best_checkpoint_metric_comp

To serve, or to finetune the trained checkpoints on a target dataset, use checkpoints under best_checkpoint_export_subdir.

PiperOrigin-RevId: 356093831

PiperOrigin-RevId: 355949726

PiperOrigin-RevId: 355946044

Imagenet, and CIFAR.

PiperOrigin-RevId: 355665447

PiperOrigin-RevId: 355637402

PiperOrigin-RevId: 355294534

PiperOrigin-RevId: 355221779

PiperOrigin-RevId: 355199395

PiperOrigin-RevId: 355088475

PiperOrigin-RevId: 355082063

PiperOrigin-RevId: 355065007

PiperOrigin-RevId: 354647274

PiperOrigin-RevId: 354439331

PiperOrigin-RevId: 354434423

PiperOrigin-RevId: 354401143