README.md
official/__init__.py
official/common/__init__.py
official/common/dataset_fn.py
official/common/distribute_utils.py
official/common/distribute_utils_test.py
official/common/flags.py
official/common/registry_imports.py
official/core/__init__.py
official/core/base_task.py
official/core/base_trainer.py
official/core/base_trainer_test.py
official/core/config_definitions.py
official/core/exp_factory.py
official/core/export_base.py
official/core/export_base_test.py
official/core/input_reader.py
official/core/registry.py
official/core/registry_test.py
official/core/task_factory.py
official/core/train_lib.py
official/core/train_lib_test.py
official/core/train_utils.py
official/core/train_utils_test.py
official/modeling/__init__.py
official/modeling/performance.py
official/modeling/tf_utils.py
official/modeling/activations/__init__.py
official/modeling/activations/gelu.py
official/modeling/activations/gelu_test.py
official/modeling/activations/relu.py
official/modeling/activations/relu_test.py
official/modeling/activations/sigmoid.py
official/modeling/activations/sigmoid_test.py
official/modeling/activations/swish.py
official/modeling/activations/swish_test.py
official/modeling/hyperparams/__init__.py
official/modeling/hyperparams/base_config.py
official/modeling/hyperparams/base_config_test.py
official/modeling/hyperparams/config_definitions.py
official/modeling/hyperparams/oneof.py
official/modeling/hyperparams/oneof_test.py
official/modeling/hyperparams/params_dict.py
official/modeling/hyperparams/params_dict_test.py
official/modeling/multitask/__init__.py
official/modeling/multitask/base_model.py
official/modeling/multitask/base_trainer.py
official/modeling/multitask/base_trainer_test.py
official/modeling/multitask/configs.py
official/modeling/multitask/evaluator.py
official/modeling/multitask/evaluator_test.py
official/modeling/multitask/interleaving_trainer.py
official/modeling/multitask/interleaving_trainer_test.py
official/modeling/multitask/multitask.py
official/modeling/multitask/task_sampler.py
official/modeling/multitask/task_sampler_test.py
official/modeling/multitask/test_utils.py
official/modeling/multitask/train_lib.py
official/modeling/multitask/train_lib_test.py
official/modeling/optimization/__init__.py
official/modeling/optimization/ema_optimizer.py
official/modeling/optimization/lr_schedule.py
official/modeling/optimization/optimizer_factory.py
official/modeling/optimization/optimizer_factory_test.py
official/modeling/optimization/configs/__init__.py
official/modeling/optimization/configs/learning_rate_config.py
official/modeling/optimization/configs/optimization_config.py
official/modeling/optimization/configs/optimization_config_test.py
official/modeling/optimization/configs/optimizer_config.py
official/nlp/__init__.py
official/nlp/continuous_finetune_lib.py
official/nlp/continuous_finetune_lib_test.py
official/nlp/optimization.py
official/nlp/train.py
official/nlp/train_ctl_continuous_finetune.py
official/nlp/albert/__init__.py
official/nlp/albert/configs.py
official/nlp/albert/run_classifier.py
official/nlp/albert/run_squad.py
official/nlp/albert/tf2_albert_encoder_checkpoint_converter.py
official/nlp/bert/__init__.py
official/nlp/bert/bert_models.py
official/nlp/bert/bert_models_test.py
official/nlp/bert/common_flags.py
official/nlp/bert/configs.py
official/nlp/bert/export_tfhub.py
official/nlp/bert/export_tfhub_test.py
official/nlp/bert/input_pipeline.py
official/nlp/bert/model_saving_utils.py
official/nlp/bert/model_training_utils.py
official/nlp/bert/model_training_utils_test.py
official/nlp/bert/run_classifier.py
official/nlp/bert/run_pretraining.py
official/nlp/bert/run_squad.py
official/nlp/bert/run_squad_helper.py
official/nlp/bert/serving.py
official/nlp/bert/squad_evaluate_v1_1.py
official/nlp/bert/squad_evaluate_v2_0.py
official/nlp/bert/tf1_checkpoint_converter_lib.py
official/nlp/bert/tf2_encoder_checkpoint_converter.py
official/nlp/bert/tokenization.py
official/nlp/bert/tokenization_test.py
official/nlp/configs/__init__.py
official/nlp/configs/bert.py
official/nlp/configs/electra.py
official/nlp/configs/encoders.py
official/nlp/configs/encoders_test.py
official/nlp/configs/experiment_configs.py
official/nlp/configs/finetuning_experiments.py
official/nlp/configs/pretraining_experiments.py
official/nlp/configs/wmt_transformer_experiments.py
official/nlp/data/__init__.py
official/nlp/data/classifier_data_lib.py
official/nlp/data/create_finetuning_data.py
official/nlp/data/create_pretraining_data.py
official/nlp/data/create_pretraining_data_test.py
official/nlp/data/create_xlnet_pretraining_data.py
official/nlp/data/create_xlnet_pretraining_data_test.py
official/nlp/data/data_loader.py
official/nlp/data/data_loader_factory.py
official/nlp/data/data_loader_factory_test.py
official/nlp/data/pretrain_dataloader.py
official/nlp/data/pretrain_dataloader_test.py
official/nlp/data/pretrain_dynamic_dataloader.py
official/nlp/data/pretrain_dynamic_dataloader_test.py
official/nlp/data/question_answering_dataloader.py
official/nlp/data/question_answering_dataloader_test.py
official/nlp/data/sentence_prediction_dataloader.py
official/nlp/data/sentence_prediction_dataloader_test.py
official/nlp/data/sentence_retrieval_lib.py
official/nlp/data/squad_lib.py
official/nlp/data/squad_lib_sp.py
official/nlp/data/tagging_data_lib.py
official/nlp/data/tagging_data_lib_test.py
official/nlp/data/tagging_dataloader.py
official/nlp/data/tagging_dataloader_test.py
official/nlp/data/train_sentencepiece.py
official/nlp/data/wmt_dataloader.py
official/nlp/data/wmt_dataloader_test.py
official/nlp/keras_nlp/__init__.py
official/nlp/keras_nlp/setup.py
official/nlp/keras_nlp/encoders/__init__.py
official/nlp/keras_nlp/encoders/bert_encoder.py
official/nlp/keras_nlp/encoders/bert_encoder_test.py
official/nlp/keras_nlp/layers/__init__.py
official/nlp/keras_nlp/layers/masked_lm.py
official/nlp/keras_nlp/layers/on_device_embedding.py
official/nlp/keras_nlp/layers/on_device_embedding_test.py
official/nlp/keras_nlp/layers/position_embedding.py
official/nlp/keras_nlp/layers/position_embedding_test.py
official/nlp/keras_nlp/layers/self_attention_mask.py
official/nlp/keras_nlp/layers/transformer_encoder_block.py
official/nlp/keras_nlp/layers/transformer_encoder_block_test.py
official/nlp/metrics/__init__.py
official/nlp/metrics/bleu.py
official/nlp/metrics/bleu_test.py
official/nlp/modeling/__init__.py
official/nlp/modeling/layers/__init__.py
official/nlp/modeling/layers/attention.py
official/nlp/modeling/layers/attention_test.py
official/nlp/modeling/layers/cls_head.py
official/nlp/modeling/layers/cls_head_test.py
official/nlp/modeling/layers/dense_einsum.py
official/nlp/modeling/layers/dense_einsum_test.py
official/nlp/modeling/layers/gated_feedforward.py
official/nlp/modeling/layers/gated_feedforward_test.py
official/nlp/modeling/layers/gaussian_process.py
official/nlp/modeling/layers/gaussian_process_test.py
official/nlp/modeling/layers/masked_lm.py
official/nlp/modeling/layers/masked_lm_test.py
official/nlp/modeling/layers/masked_softmax.py
official/nlp/modeling/layers/masked_softmax_test.py
official/nlp/modeling/layers/mat_mul_with_margin.py
official/nlp/modeling/layers/mat_mul_with_margin_test.py
official/nlp/modeling/layers/mobile_bert_layers.py
official/nlp/modeling/layers/mobile_bert_layers_test.py
official/nlp/modeling/layers/multi_channel_attention.py
official/nlp/modeling/layers/multi_channel_attention_test.py
official/nlp/modeling/layers/on_device_embedding.py
official/nlp/modeling/layers/position_embedding.py
official/nlp/modeling/layers/position_embedding_test.py
official/nlp/modeling/layers/relative_attention.py
official/nlp/modeling/layers/relative_attention_test.py
official/nlp/modeling/layers/rezero_transformer.py
official/nlp/modeling/layers/rezero_transformer_test.py
official/nlp/modeling/layers/self_attention_mask.py
official/nlp/modeling/layers/spectral_normalization.py
official/nlp/modeling/layers/spectral_normalization_test.py
official/nlp/modeling/layers/talking_heads_attention.py
official/nlp/modeling/layers/talking_heads_attention_test.py
official/nlp/modeling/layers/text_layers.py
official/nlp/modeling/layers/text_layers_test.py
official/nlp/modeling/layers/tn_expand_condense.py
official/nlp/modeling/layers/tn_expand_condense_test.py
official/nlp/modeling/layers/tn_transformer_expand_condense.py
official/nlp/modeling/layers/tn_transformer_test.py
official/nlp/modeling/layers/transformer.py
official/nlp/modeling/layers/transformer_scaffold.py
official/nlp/modeling/layers/transformer_scaffold_test.py
official/nlp/modeling/layers/transformer_test.py
official/nlp/modeling/layers/transformer_xl.py
official/nlp/modeling/layers/transformer_xl_test.py
official/nlp/modeling/layers/util.py
official/nlp/modeling/losses/__init__.py
official/nlp/modeling/losses/weighted_sparse_categorical_crossentropy.py
official/nlp/modeling/losses/weighted_sparse_categorical_crossentropy_test.py
official/nlp/modeling/models/__init__.py
official/nlp/modeling/models/bert_classifier.py
official/nlp/modeling/models/bert_classifier_test.py
official/nlp/modeling/models/bert_pretrainer.py
official/nlp/modeling/models/bert_pretrainer_test.py
official/nlp/modeling/models/bert_span_labeler.py
official/nlp/modeling/models/bert_span_labeler_test.py
official/nlp/modeling/models/bert_token_classifier.py
official/nlp/modeling/models/bert_token_classifier_test.py
official/nlp/modeling/models/dual_encoder.py
official/nlp/modeling/models/dual_encoder_test.py
official/nlp/modeling/models/electra_pretrainer.py
official/nlp/modeling/models/electra_pretrainer_test.py
official/nlp/modeling/models/seq2seq_transformer.py
official/nlp/modeling/models/seq2seq_transformer_test.py
official/nlp/modeling/models/xlnet.py
official/nlp/modeling/models/xlnet_test.py
official/nlp/modeling/networks/__init__.py
official/nlp/modeling/networks/albert_encoder.py
official/nlp/modeling/networks/albert_encoder_test.py
official/nlp/modeling/networks/bert_encoder.py
official/nlp/modeling/networks/bert_encoder_test.py
official/nlp/modeling/networks/classification.py
official/nlp/modeling/networks/classification_test.py
official/nlp/modeling/networks/encoder_scaffold.py
official/nlp/modeling/networks/encoder_scaffold_test.py
official/nlp/modeling/networks/mobile_bert_encoder.py
official/nlp/modeling/networks/mobile_bert_encoder_test.py
official/nlp/modeling/networks/packed_sequence_embedding.py
official/nlp/modeling/networks/packed_sequence_embedding_test.py
official/nlp/modeling/networks/span_labeling.py
official/nlp/modeling/networks/span_labeling_test.py
official/nlp/modeling/networks/xlnet_base.py
official/nlp/modeling/networks/xlnet_base_test.py
official/nlp/modeling/ops/__init__.py
official/nlp/modeling/ops/beam_search.py
official/nlp/modeling/ops/beam_search_test.py
official/nlp/modeling/ops/decoding_module.py
official/nlp/modeling/ops/decoding_module_test.py
official/nlp/modeling/ops/sampling_module.py
official/nlp/modeling/ops/segment_extractor.py
official/nlp/modeling/ops/segment_extractor_test.py
official/nlp/nhnet/__init__.py
official/nlp/nhnet/configs.py
official/nlp/nhnet/configs_test.py
official/nlp/nhnet/decoder.py
official/nlp/nhnet/decoder_test.py
official/nlp/nhnet/evaluation.py
official/nlp/nhnet/input_pipeline.py
official/nlp/nhnet/models.py
official/nlp/nhnet/models_test.py
official/nlp/nhnet/optimizer.py
official/nlp/nhnet/raw_data_process.py
official/nlp/nhnet/raw_data_processor.py
official/nlp/nhnet/trainer.py
official/nlp/nhnet/trainer_test.py
official/nlp/nhnet/utils.py
official/nlp/projects/__init__.py
official/nlp/projects/bigbird/__init__.py
official/nlp/projects/bigbird/attention.py
official/nlp/projects/bigbird/attention_test.py
official/nlp/projects/bigbird/encoder.py
official/nlp/projects/bigbird/encoder_test.py
official/nlp/projects/triviaqa/__init__.py
official/nlp/projects/triviaqa/dataset.py
official/nlp/projects/triviaqa/download_and_prepare.py
official/nlp/projects/triviaqa/evaluate.py
official/nlp/projects/triviaqa/evaluation.py
official/nlp/projects/triviaqa/inputs.py
official/nlp/projects/triviaqa/modeling.py
official/nlp/projects/triviaqa/predict.py
official/nlp/projects/triviaqa/prediction.py
official/nlp/projects/triviaqa/preprocess.py
official/nlp/projects/triviaqa/sentencepiece_pb2.py
official/nlp/projects/triviaqa/train.py
official/nlp/tasks/__init__.py
official/nlp/tasks/electra_task.py
official/nlp/tasks/electra_task_test.py
official/nlp/tasks/masked_lm.py
official/nlp/tasks/masked_lm_test.py
official/nlp/tasks/question_answering.py
official/nlp/tasks/question_answering_test.py
official/nlp/tasks/sentence_prediction.py
official/nlp/tasks/sentence_prediction_test.py
official/nlp/tasks/tagging.py
official/nlp/tasks/tagging_test.py
official/nlp/tasks/translation.py
official/nlp/tasks/translation_test.py
official/nlp/tasks/utils.py
official/nlp/tools/__init__.py
official/nlp/tools/export_tfhub.py
official/nlp/tools/export_tfhub_lib.py
official/nlp/tools/export_tfhub_lib_test.py
official/nlp/transformer/__init__.py
official/nlp/transformer/attention_layer.py
official/nlp/transformer/beam_search_v1.py
official/nlp/transformer/compute_bleu.py
official/nlp/transformer/compute_bleu_test.py
official/nlp/transformer/data_download.py
official/nlp/transformer/data_pipeline.py
official/nlp/transformer/embedding_layer.py
official/nlp/transformer/ffn_layer.py
official/nlp/transformer/metrics.py
official/nlp/transformer/misc.py
official/nlp/transformer/model_params.py
official/nlp/transformer/model_utils.py
official/nlp/transformer/model_utils_test.py
official/nlp/transformer/optimizer.py
official/nlp/transformer/transformer.py
official/nlp/transformer/transformer_forward_test.py
official/nlp/transformer/transformer_layers_test.py
official/nlp/transformer/transformer_main.py
official/nlp/transformer/transformer_main_test.py
official/nlp/transformer/transformer_test.py
official/nlp/transformer/translate.py
official/nlp/transformer/utils/__init__.py
official/nlp/transformer/utils/metrics.py
official/nlp/transformer/utils/tokenizer.py
official/nlp/transformer/utils/tokenizer_test.py
official/nlp/xlnet/__init__.py
official/nlp/xlnet/classifier_utils.py
official/nlp/xlnet/common_flags.py
official/nlp/xlnet/data_utils.py
official/nlp/xlnet/optimization.py
official/nlp/xlnet/preprocess_classification_data.py
official/nlp/xlnet/preprocess_pretrain_data.py
official/nlp/xlnet/preprocess_squad_data.py
official/nlp/xlnet/preprocess_utils.py
official/nlp/xlnet/run_classifier.py
official/nlp/xlnet/run_pretrain.py
official/nlp/xlnet/run_squad.py
official/nlp/xlnet/squad_utils.py
official/nlp/xlnet/training_utils.py
official/nlp/xlnet/xlnet_config.py
official/nlp/xlnet/xlnet_modeling.py
official/recommendation/__init__.py
official/recommendation/constants.py
official/recommendation/create_ncf_data.py
official/recommendation/data_pipeline.py
official/recommendation/data_preprocessing.py
official/recommendation/data_test.py
official/recommendation/movielens.py
official/recommendation/ncf_common.py
official/recommendation/ncf_input_pipeline.py
official/recommendation/ncf_keras_main.py
official/recommendation/ncf_test.py
official/recommendation/neumf_model.py
official/recommendation/popen_helper.py
official/recommendation/stat_utils.py
official/recommendation/ranking/__init__.py
official/recommendation/ranking/common.py
official/recommendation/ranking/data_pipeline.py
official/recommendation/ranking/data_pipeline_test.py
official/recommendation/ranking/task.py
official/recommendation/ranking/task_test.py
official/recommendation/ranking/train.py
official/recommendation/ranking/train_test.py
official/staging/__init__.py
official/staging/training/__init__.py
official/staging/training/grad_utils.py
official/utils/__init__.py
official/utils/hyperparams_flags.py
official/utils/flags/__init__.py
official/utils/flags/_base.py
official/utils/flags/_benchmark.py
official/utils/flags/_conventions.py
official/utils/flags/_device.py
official/utils/flags/_distribution.py
official/utils/flags/_misc.py
official/utils/flags/_performance.py
official/utils/flags/core.py
official/utils/flags/flags_test.py
official/utils/misc/__init__.py
official/utils/misc/distribution_utils.py
official/utils/misc/keras_utils.py
official/utils/misc/model_helpers.py
official/utils/misc/model_helpers_test.py
official/utils/testing/__init__.py
official/utils/testing/integration.py
official/utils/testing/mock_task.py
official/vision/__init__.py
official/vision/beta/__init__.py
official/vision/beta/train.py
official/vision/beta/train_spatial_partitioning.py
official/vision/beta/configs/__init__.py
official/vision/beta/configs/backbones.py
official/vision/beta/configs/backbones_3d.py
official/vision/beta/configs/common.py
official/vision/beta/configs/decoders.py
official/vision/beta/configs/image_classification.py
official/vision/beta/configs/image_classification_test.py
official/vision/beta/configs/maskrcnn.py
official/vision/beta/configs/retinanet.py
official/vision/beta/configs/semantic_segmentation.py
official/vision/beta/configs/semantic_segmentation_test.py
official/vision/beta/configs/video_classification.py
official/vision/beta/configs/video_classification_test.py
official/vision/beta/data/__init__.py
official/vision/beta/data/create_coco_tf_record.py
official/vision/beta/data/tfrecord_lib.py
official/vision/beta/data/tfrecord_lib_test.py
official/vision/beta/dataloaders/__init__.py
official/vision/beta/dataloaders/classification_input.py
official/vision/beta/dataloaders/decoder.py
official/vision/beta/dataloaders/input_reader.py
official/vision/beta/dataloaders/input_reader_factory.py
official/vision/beta/dataloaders/maskrcnn_input.py
official/vision/beta/dataloaders/parser.py
official/vision/beta/dataloaders/retinanet_input.py
official/vision/beta/dataloaders/segmentation_input.py
official/vision/beta/dataloaders/tf_example_decoder.py
official/vision/beta/dataloaders/tf_example_decoder_test.py
official/vision/beta/dataloaders/tf_example_label_map_decoder.py
official/vision/beta/dataloaders/tf_example_label_map_decoder_test.py
official/vision/beta/dataloaders/tfds_classification_decoders.py
official/vision/beta/dataloaders/tfds_detection_decoders.py
official/vision/beta/dataloaders/tfds_segmentation_decoders.py
official/vision/beta/dataloaders/utils.py
official/vision/beta/dataloaders/video_input.py
official/vision/beta/dataloaders/video_input_test.py
official/vision/beta/evaluation/__init__.py
official/vision/beta/evaluation/coco_evaluator.py
official/vision/beta/evaluation/coco_utils.py
official/vision/beta/evaluation/segmentation_metrics.py
official/vision/beta/losses/__init__.py
official/vision/beta/losses/maskrcnn_losses.py
official/vision/beta/losses/retinanet_losses.py
official/vision/beta/losses/segmentation_losses.py
official/vision/beta/modeling/__init__.py
official/vision/beta/modeling/classification_model.py
official/vision/beta/modeling/classification_model_test.py
official/vision/beta/modeling/factory.py
official/vision/beta/modeling/factory_3d.py
official/vision/beta/modeling/factory_test.py
official/vision/beta/modeling/maskrcnn_model.py
official/vision/beta/modeling/maskrcnn_model_test.py
official/vision/beta/modeling/retinanet_model.py
official/vision/beta/modeling/retinanet_model_test.py
official/vision/beta/modeling/segmentation_model.py
official/vision/beta/modeling/segmentation_model_test.py
official/vision/beta/modeling/video_classification_model.py
official/vision/beta/modeling/video_classification_model_test.py
official/vision/beta/modeling/backbones/__init__.py
official/vision/beta/modeling/backbones/efficientnet.py
official/vision/beta/modeling/backbones/efficientnet_test.py
official/vision/beta/modeling/backbones/factory.py
official/vision/beta/modeling/backbones/factory_test.py
official/vision/beta/modeling/backbones/mobilenet.py
official/vision/beta/modeling/backbones/mobilenet_test.py
official/vision/beta/modeling/backbones/resnet.py
official/vision/beta/modeling/backbones/resnet_3d.py
official/vision/beta/modeling/backbones/resnet_3d_test.py
official/vision/beta/modeling/backbones/resnet_deeplab.py
official/vision/beta/modeling/backbones/resnet_deeplab_test.py
official/vision/beta/modeling/backbones/resnet_test.py
official/vision/beta/modeling/backbones/revnet.py
official/vision/beta/modeling/backbones/revnet_test.py
official/vision/beta/modeling/backbones/spinenet.py
official/vision/beta/modeling/backbones/spinenet_mobile.py
official/vision/beta/modeling/backbones/spinenet_mobile_test.py
official/vision/beta/modeling/backbones/spinenet_test.py
official/vision/beta/modeling/decoders/__init__.py
official/vision/beta/modeling/decoders/aspp.py
official/vision/beta/modeling/decoders/aspp_test.py
official/vision/beta/modeling/decoders/factory.py
official/vision/beta/modeling/decoders/fpn.py
official/vision/beta/modeling/decoders/fpn_test.py
official/vision/beta/modeling/decoders/nasfpn.py
official/vision/beta/modeling/decoders/nasfpn_test.py
official/vision/beta/modeling/heads/__init__.py
official/vision/beta/modeling/heads/dense_prediction_heads.py
official/vision/beta/modeling/heads/dense_prediction_heads_test.py
official/vision/beta/modeling/heads/instance_heads.py
official/vision/beta/modeling/heads/instance_heads_test.py
official/vision/beta/modeling/heads/segmentation_heads.py
official/vision/beta/modeling/heads/segmentation_heads_test.py
official/vision/beta/modeling/layers/__init__.py
official/vision/beta/modeling/layers/box_sampler.py
official/vision/beta/modeling/layers/detection_generator.py
official/vision/beta/modeling/layers/detection_generator_test.py
official/vision/beta/modeling/layers/mask_sampler.py
official/vision/beta/modeling/layers/nn_blocks.py
official/vision/beta/modeling/layers/nn_blocks_3d.py
official/vision/beta/modeling/layers/nn_blocks_3d_test.py
official/vision/beta/modeling/layers/nn_blocks_test.py
official/vision/beta/modeling/layers/nn_layers.py
official/vision/beta/modeling/layers/nn_layers_test.py
official/vision/beta/modeling/layers/roi_aligner.py
official/vision/beta/modeling/layers/roi_aligner_test.py
official/vision/beta/modeling/layers/roi_generator.py
official/vision/beta/modeling/layers/roi_sampler.py
official/vision/beta/ops/__init__.py
official/vision/beta/ops/anchor.py
official/vision/beta/ops/anchor_test.py
official/vision/beta/ops/augment.py
official/vision/beta/ops/augment_test.py
official/vision/beta/ops/box_ops.py
official/vision/beta/ops/mask_ops.py
official/vision/beta/ops/mask_ops_test.py
official/vision/beta/ops/nms.py
official/vision/beta/ops/preprocess_ops.py
official/vision/beta/ops/preprocess_ops_3d.py
official/vision/beta/ops/preprocess_ops_3d_test.py
official/vision/beta/ops/preprocess_ops_test.py
official/vision/beta/ops/sampling_ops.py
official/vision/beta/ops/spatial_transform_ops.py
official/vision/beta/projects/__init__.py
official/vision/beta/projects/yt8m/__init__.py
official/vision/beta/projects/yt8m/train.py
official/vision/beta/projects/yt8m/configs/__init__.py
official/vision/beta/projects/yt8m/configs/yt8m.py
official/vision/beta/projects/yt8m/modeling/__init__.py
official/vision/beta/projects/yt8m/modeling/yt8m_agg_models.py
official/vision/beta/projects/yt8m/modeling/yt8m_model.py
official/vision/beta/projects/yt8m/modeling/yt8m_model_test.py
official/vision/beta/projects/yt8m/modeling/yt8m_model_utils.py
official/vision/beta/projects/yt8m/tasks/__init__.py
official/vision/beta/projects/yt8m/tasks/yt8m_task.py
official/vision/beta/serving/__init__.py
official/vision/beta/serving/detection.py
official/vision/beta/serving/detection_test.py
official/vision/beta/serving/export_base.py
official/vision/beta/serving/export_saved_model.py
official/vision/beta/serving/export_saved_model_lib.py
official/vision/beta/serving/export_tfhub.py
official/vision/beta/serving/image_classification.py
official/vision/beta/serving/image_classification_test.py
official/vision/beta/serving/semantic_segmentation.py
official/vision/beta/serving/semantic_segmentation_test.py
official/vision/beta/tasks/__init__.py
official/vision/beta/tasks/image_classification.py
official/vision/beta/tasks/maskrcnn.py
official/vision/beta/tasks/retinanet.py
official/vision/beta/tasks/semantic_segmentation.py
official/vision/beta/tasks/video_classification.py
official/vision/detection/__init__.py
official/vision/detection/main.py
official/vision/detection/configs/__init__.py
official/vision/detection/configs/base_config.py
official/vision/detection/configs/factory.py
official/vision/detection/configs/maskrcnn_config.py
official/vision/detection/configs/olnmask_config.py
official/vision/detection/configs/retinanet_config.py
official/vision/detection/configs/shapemask_config.py
official/vision/detection/dataloader/__init__.py
official/vision/detection/dataloader/anchor.py
official/vision/detection/dataloader/factory.py
official/vision/detection/dataloader/input_reader.py
official/vision/detection/dataloader/maskrcnn_parser.py
official/vision/detection/dataloader/mode_keys.py
official/vision/detection/dataloader/olnmask_parser.py
official/vision/detection/dataloader/retinanet_parser.py
official/vision/detection/dataloader/shapemask_parser.py
official/vision/detection/dataloader/tf_example_decoder.py
official/vision/detection/evaluation/__init__.py
official/vision/detection/evaluation/coco_evaluator.py
official/vision/detection/evaluation/coco_utils.py
official/vision/detection/evaluation/factory.py
official/vision/detection/executor/__init__.py
official/vision/detection/executor/detection_executor.py
official/vision/detection/executor/distributed_executor.py
official/vision/detection/modeling/__init__.py
official/vision/detection/modeling/base_model.py
official/vision/detection/modeling/checkpoint_utils.py
official/vision/detection/modeling/factory.py
official/vision/detection/modeling/learning_rates.py
official/vision/detection/modeling/losses.py
official/vision/detection/modeling/maskrcnn_model.py
official/vision/detection/modeling/olnmask_model.py
official/vision/detection/modeling/optimizers.py
official/vision/detection/modeling/retinanet_model.py
official/vision/detection/modeling/shapemask_model.py
official/vision/detection/modeling/architecture/__init__.py
official/vision/detection/modeling/architecture/factory.py
official/vision/detection/modeling/architecture/fpn.py
official/vision/detection/modeling/architecture/heads.py
official/vision/detection/modeling/architecture/identity.py
official/vision/detection/modeling/architecture/nn_blocks.py
official/vision/detection/modeling/architecture/nn_ops.py
official/vision/detection/modeling/architecture/resnet.py
official/vision/detection/modeling/architecture/spinenet.py
official/vision/detection/ops/__init__.py
official/vision/detection/ops/nms.py
official/vision/detection/ops/postprocess_ops.py
official/vision/detection/ops/roi_ops.py
official/vision/detection/ops/spatial_transform_ops.py
official/vision/detection/ops/target_ops.py
official/vision/detection/utils/__init__.py
official/vision/detection/utils/box_utils.py
official/vision/detection/utils/class_utils.py
official/vision/detection/utils/dataloader_utils.py
official/vision/detection/utils/input_utils.py
official/vision/detection/utils/mask_utils.py
official/vision/detection/utils/object_detection/__init__.py
official/vision/detection/utils/object_detection/argmax_matcher.py
official/vision/detection/utils/object_detection/balanced_positive_negative_sampler.py
official/vision/detection/utils/object_detection/box_coder.py
official/vision/detection/utils/object_detection/box_list.py
official/vision/detection/utils/object_detection/box_list_ops.py
official/vision/detection/utils/object_detection/faster_rcnn_box_coder.py
official/vision/detection/utils/object_detection/matcher.py
official/vision/detection/utils/object_detection/minibatch_sampler.py
official/vision/detection/utils/object_detection/ops.py
official/vision/detection/utils/object_detection/preprocessor.py
official/vision/detection/utils/object_detection/region_similarity_calculator.py
official/vision/detection/utils/object_detection/shape_utils.py
official/vision/detection/utils/object_detection/target_assigner.py
official/vision/detection/utils/object_detection/visualization_utils.py
official/vision/image_classification/__init__.py
official/vision/image_classification/augment.py
official/vision/image_classification/augment_test.py
official/vision/image_classification/callbacks.py
official/vision/image_classification/classifier_trainer.py
official/vision/image_classification/classifier_trainer_test.py
official/vision/image_classification/classifier_trainer_util_test.py
official/vision/image_classification/dataset_factory.py
official/vision/image_classification/learning_rate.py
official/vision/image_classification/learning_rate_test.py
official/vision/image_classification/mnist_main.py
official/vision/image_classification/mnist_test.py
official/vision/image_classification/optimizer_factory.py
official/vision/image_classification/optimizer_factory_test.py
official/vision/image_classification/preprocessing.py
official/vision/image_classification/test_utils.py
official/vision/image_classification/configs/__init__.py
official/vision/image_classification/configs/base_configs.py
official/vision/image_classification/configs/configs.py
official/vision/image_classification/efficientnet/__init__.py
official/vision/image_classification/efficientnet/common_modules.py
official/vision/image_classification/efficientnet/efficientnet_config.py
official/vision/image_classification/efficientnet/efficientnet_model.py
official/vision/image_classification/efficientnet/tfhub_export.py
official/vision/image_classification/resnet/__init__.py
official/vision/image_classification/resnet/common.py
official/vision/image_classification/resnet/imagenet_preprocessing.py
official/vision/image_classification/resnet/resnet_config.py
official/vision/image_classification/resnet/resnet_ctl_imagenet_main.py
official/vision/image_classification/resnet/resnet_model.py
official/vision/image_classification/resnet/resnet_runnable.py
official/vision/image_classification/resnet/tfhub_export.py
official/vision/keras_cv/__init__.py
official/vision/keras_cv/setup.py
official/vision/keras_cv/layers/__init__.py
official/vision/keras_cv/layers/deeplab.py
official/vision/keras_cv/layers/deeplab_test.py
official/vision/keras_cv/losses/__init__.py
official/vision/keras_cv/losses/focal_loss.py
official/vision/keras_cv/losses/loss_utils.py
official/vision/keras_cv/metrics/__init__.py
official/vision/keras_cv/metrics/iou.py
official/vision/keras_cv/metrics/iou_test.py
official/vision/keras_cv/ops/__init__.py
official/vision/keras_cv/ops/anchor_generator.py
official/vision/keras_cv/ops/anchor_generator_test.py
official/vision/keras_cv/ops/box_matcher.py
official/vision/keras_cv/ops/box_matcher_test.py
official/vision/keras_cv/ops/iou_similarity.py
official/vision/keras_cv/ops/iou_similarity_test.py
official/vision/keras_cv/ops/target_gather.py
official/vision/keras_cv/ops/target_gather_test.py
orbit/__init__.py
orbit/controller.py
orbit/controller_test.py
orbit/runner.py
orbit/standard_runner.py
orbit/standard_runner_test.py
orbit/examples/__init__.py
orbit/examples/single_task/__init__.py
orbit/examples/single_task/single_task_evaluator.py
orbit/examples/single_task/single_task_evaluator_test.py
orbit/examples/single_task/single_task_trainer.py
orbit/examples/single_task/single_task_trainer_test.py
orbit/utils/__init__.py
orbit/utils/common.py
orbit/utils/common_test.py
orbit/utils/epoch_helper.py
orbit/utils/loop_fns.py
orbit/utils/summary_manager.py
orbit/utils/tpu_summaries.py
orbit/utils/tpu_summaries_test.py
tf_models_official.egg-info/PKG-INFO
tf_models_official.egg-info/SOURCES.txt
tf_models_official.egg-info/dependency_links.txt
tf_models_official.egg-info/requires.txt
tf_models_official.egg-info/top_level.txt