Path	Lines of Code
official/projects/__init__.py	1
official/projects/assemblenet/experiment/ucf101_assemblenet_plus_tpu.yaml	53
official/projects/assemblenet/experiment/ucf101_assemblenet_tpu.yaml	52
official/projects/assemblenet/modeling/assemblenet.py	671
official/projects/assemblenet/modeling/assemblenet_plus.py	470
official/projects/assemblenet/modeling/rep_flow_2d_layer.py	276
official/projects/assemblenet/train.py	71
official/projects/basnet/configs/basnet.py	118
official/projects/basnet/configs/experiments/basnet_dut_gpu.yaml	10
official/projects/basnet/evaluation/metrics.py	171
official/projects/basnet/losses/basnet_losses.py	35
official/projects/basnet/modeling/basnet_model.py	288
official/projects/basnet/modeling/nn_blocks.py	158
official/projects/basnet/modeling/refunet.py	106
official/projects/basnet/serving/basnet.py	31
official/projects/basnet/serving/export_saved_model.py	59
official/projects/basnet/tasks/basnet.py	171
official/projects/basnet/train.py	10
official/projects/bigbird/__init__.py	1
official/projects/bigbird/encoder.py	146
official/projects/bigbird/experiment_configs.py	80
official/projects/bigbird/experiments/glue_mnli_matched.yaml	49
official/projects/bigbird/experiments/squad_v1.yaml	55
official/projects/bigbird/recompute_grad.py	123
official/projects/bigbird/recomputing_dropout.py	74
official/projects/bigbird/stateless_dropout.py	58
official/projects/detr/configs/detr.py	77
official/projects/detr/dataloaders/coco.py	117
official/projects/detr/modeling/detr.py	191
official/projects/detr/modeling/transformer.py	582
official/projects/detr/ops/matchers.py	192
official/projects/detr/optimization.py	99
official/projects/detr/tasks/detection.py	219
official/projects/detr/train.py	38
official/projects/edgetpu/nlp/__init__.py	1
official/projects/edgetpu/nlp/configs/__init__.py	1
official/projects/edgetpu/nlp/configs/params.py	74
official/projects/edgetpu/nlp/experiments/downstream_tasks/glue_mnli.yaml	44
official/projects/edgetpu/nlp/experiments/downstream_tasks/mobilebert_baseline.yaml	22
official/projects/edgetpu/nlp/experiments/downstream_tasks/mobilebert_edgetpu_m.yaml	22
official/projects/edgetpu/nlp/experiments/downstream_tasks/mobilebert_edgetpu_s.yaml	22
official/projects/edgetpu/nlp/experiments/downstream_tasks/mobilebert_edgetpu_xs.yaml	22
official/projects/edgetpu/nlp/experiments/downstream_tasks/squad_v1.yaml	49
official/projects/edgetpu/nlp/experiments/mobilebert_baseline.yaml	141
official/projects/edgetpu/nlp/experiments/mobilebert_edgetpu_m.yaml	141
official/projects/edgetpu/nlp/experiments/mobilebert_edgetpu_s.yaml	141
official/projects/edgetpu/nlp/experiments/mobilebert_edgetpu_xs.yaml	141
official/projects/edgetpu/nlp/mobilebert_edgetpu_trainer.py	382
official/projects/edgetpu/nlp/modeling/__init__.py	1
official/projects/edgetpu/nlp/modeling/edgetpu_layers.py	78
official/projects/edgetpu/nlp/modeling/encoder.py	115
official/projects/edgetpu/nlp/modeling/model_builder.py	58
official/projects/edgetpu/nlp/modeling/pretrainer.py	76
official/projects/edgetpu/nlp/run_mobilebert_edgetpu_train.py	77
official/projects/edgetpu/nlp/serving/__init__.py	1
official/projects/edgetpu/nlp/serving/export_tflite_squad.py	108
official/projects/edgetpu/nlp/utils/__init__.py	1
official/projects/edgetpu/nlp/utils/utils.py	49
official/projects/edgetpu/vision/__init__.py	1
official/projects/edgetpu/vision/configs/__init__.py	1
official/projects/edgetpu/vision/configs/mobilenet_edgetpu_config.py	142
official/projects/edgetpu/vision/configs/semantic_segmentation_config.py	183
official/projects/edgetpu/vision/configs/semantic_segmentation_searched_config.py	147
official/projects/edgetpu/vision/dataloaders/__init__.py	1
official/projects/edgetpu/vision/dataloaders/classification_input.py	88
official/projects/edgetpu/vision/modeling/__init__.py	1
official/projects/edgetpu/vision/modeling/backbones/__init__.py	1
official/projects/edgetpu/vision/modeling/backbones/mobilenet_edgetpu.py	70
official/projects/edgetpu/vision/modeling/common_modules.py	91
official/projects/edgetpu/vision/modeling/custom_layers.py	268
official/projects/edgetpu/vision/modeling/heads/__init__.py	1
official/projects/edgetpu/vision/modeling/heads/bifpn_head.py	557
official/projects/edgetpu/vision/modeling/mobilenet_edgetpu_v1_model.py	65
official/projects/edgetpu/vision/modeling/mobilenet_edgetpu_v1_model_blocks.py	289
official/projects/edgetpu/vision/modeling/mobilenet_edgetpu_v2_model.py	98
official/projects/edgetpu/vision/modeling/mobilenet_edgetpu_v2_model_blocks.py	857
official/projects/edgetpu/vision/serving/__init__.py	1
official/projects/edgetpu/vision/serving/export_tflite.py	127
official/projects/edgetpu/vision/serving/export_util.py	131
official/projects/edgetpu/vision/serving/tflite_imagenet_evaluator.py	54
official/projects/edgetpu/vision/serving/tflite_imagenet_evaluator_run.py	39
official/projects/edgetpu/vision/tasks/__init__.py	1
official/projects/edgetpu/vision/tasks/image_classification.py	229
official/projects/edgetpu/vision/tasks/semantic_segmentation.py	226
official/projects/edgetpu/vision/train.py	43
official/projects/mobilebert/__init__.py	1
official/projects/mobilebert/distillation.py	409
official/projects/mobilebert/experiments/en_uncased_student.yaml	22
official/projects/mobilebert/experiments/en_uncased_teacher.yaml	22
official/projects/mobilebert/experiments/mobilebert_distillation_en_uncased.yaml	79
official/projects/mobilebert/export_tfhub.py	50
official/projects/mobilebert/model_utils.py	99
official/projects/mobilebert/run_distillation.py	91
official/projects/mobilebert/tf2_model_checkpoint_converter.py	174
official/projects/mobilebert/utils.py	10
official/projects/movinet/__init__.py	1
official/projects/movinet/configs/__init__.py	1
official/projects/movinet/configs/movinet.py	62
official/projects/movinet/configs/yaml/movinet_a0_k600_8x8.yaml	69
official/projects/movinet/configs/yaml/movinet_a0_k600_cpu_local.yaml	56
official/projects/movinet/configs/yaml/movinet_a0_stream_k600_8x8.yaml	73
official/projects/movinet/configs/yaml/movinet_a1_k600_8x8.yaml	69
official/projects/movinet/configs/yaml/movinet_a1_stream_k600_8x8.yaml	73
official/projects/movinet/configs/yaml/movinet_a2_k600_8x8.yaml	69
official/projects/movinet/configs/yaml/movinet_a2_stream_k600_8x8.yaml	73
official/projects/movinet/configs/yaml/movinet_a3_k600_8x8.yaml	69
official/projects/movinet/configs/yaml/movinet_a3_stream_k600_8x8.yaml	74
official/projects/movinet/configs/yaml/movinet_a4_k600_8x8.yaml	69
official/projects/movinet/configs/yaml/movinet_a4_stream_k600_8x8.yaml	75
official/projects/movinet/configs/yaml/movinet_a5_k600_8x8.yaml	69
official/projects/movinet/configs/yaml/movinet_a5_stream_k600_8x8.yaml	75
official/projects/movinet/configs/yaml/movinet_t0_k600_8x8.yaml	68
official/projects/movinet/configs/yaml/movinet_t0_stream_k600_8x8.yaml	73
official/projects/movinet/modeling/__init__.py	1
official/projects/movinet/modeling/movinet.py	568
official/projects/movinet/modeling/movinet_layers.py	1014
official/projects/movinet/modeling/movinet_model.py	161
official/projects/movinet/tools/convert_3d_2plus1d.py	76
official/projects/movinet/tools/export_saved_model.py	127
official/projects/movinet/train.py	43
official/projects/nhnet/__init__.py	1
official/projects/nhnet/configs.py	63
official/projects/nhnet/decoder.py	269
official/projects/nhnet/evaluation.py	97
official/projects/nhnet/input_pipeline.py	162
official/projects/nhnet/models.py	436
official/projects/nhnet/optimizer.py	30
official/projects/nhnet/raw_data_process.py	60
official/projects/nhnet/raw_data_processor.py	169
official/projects/nhnet/trainer.py	177
official/projects/nhnet/utils.py	49
official/projects/roformer/__init__.py	1
official/projects/roformer/experiments/roformer_base.yaml	61
official/projects/roformer/roformer.py	25
official/projects/roformer/roformer_attention.py	70
official/projects/roformer/roformer_encoder.py	167
official/projects/roformer/roformer_encoder_block.py	223
official/projects/roformer/roformer_experiments.py	96
official/projects/roformer/train.py	37
official/projects/teams/__init__.py	1
official/projects/teams/experiments/base/glue_mnli.yaml	39
official/projects/teams/experiments/base/squad_v1.yaml	47
official/projects/teams/experiments/base/squad_v2.yaml	48
official/projects/teams/experiments/base/wiki_books_pretrain.yaml	75
official/projects/teams/experiments/small/glue_mnli.yaml	38
official/projects/teams/experiments/small/squad_v1.yaml	46
official/projects/teams/experiments/small/squad_v2.yaml	47
official/projects/teams/experiments/small/wiki_books_pretrain.yaml	75
official/projects/teams/experiments/teams_en_uncased_base.yaml	17
official/projects/teams/experiments/teams_en_uncased_small.yaml	17
official/projects/teams/teams.py	61
official/projects/teams/teams_experiments.py	79
official/projects/teams/teams_pretrainer.py	258
official/projects/teams/teams_task.py	175
official/projects/text_classification_example/classification_data_loader.py	45
official/projects/text_classification_example/classification_example.py	127
official/projects/text_classification_example/experiments/classification_ft_cola.yaml	41
official/projects/text_classification_example/experiments/local_example.yaml	40
official/projects/text_classification_example/train.py	36
official/projects/triviaqa/__init__.py	1
official/projects/triviaqa/dataset.py	333
official/projects/triviaqa/download_and_prepare.py	44
official/projects/triviaqa/evaluate.py	24
official/projects/triviaqa/evaluation.py	119
official/projects/triviaqa/inputs.py	308
official/projects/triviaqa/modeling.py	78
official/projects/triviaqa/predict.py	139
official/projects/triviaqa/prediction.py	40
official/projects/triviaqa/preprocess.py	427
official/projects/triviaqa/sentencepiece_pb2.py	279
official/projects/triviaqa/train.py	314
official/projects/vit/configs/__init__.py	1
official/projects/vit/configs/backbones.py	24
official/projects/vit/configs/image_classification.py	235
official/projects/vit/modeling/nn_blocks.py	72
official/projects/vit/modeling/vit.py	221
official/projects/vit/train.py	8
official/projects/volumetric_models/configs/backbones.py	14
official/projects/volumetric_models/configs/decoders.py	18
official/projects/volumetric_models/configs/semantic_segmentation_3d.py	125
official/projects/volumetric_models/dataloaders/segmentation_input_3d.py	56
official/projects/volumetric_models/evaluation/segmentation_metrics.py	63
official/projects/volumetric_models/losses/segmentation_losses.py	51
official/projects/volumetric_models/modeling/backbones/__init__.py	1
official/projects/volumetric_models/modeling/backbones/unet_3d.py	106
official/projects/volumetric_models/modeling/decoders/__init__.py	1
official/projects/volumetric_models/modeling/decoders/factory.py	26
official/projects/volumetric_models/modeling/decoders/unet_3d_decoder.py	111
official/projects/volumetric_models/modeling/factory.py	36
official/projects/volumetric_models/modeling/heads/segmentation_heads_3d.py	104
official/projects/volumetric_models/modeling/nn_blocks_3d.py	357
official/projects/volumetric_models/registry_imports.py	5
official/projects/volumetric_models/serving/export_saved_model.py	74
official/projects/volumetric_models/serving/semantic_segmentation_3d.py	22
official/projects/volumetric_models/tasks/semantic_segmentation_3d.py	204
official/projects/volumetric_models/train.py	10
official/projects/yt8m/__init__.py	1
official/projects/yt8m/configs/__init__.py	1
official/projects/yt8m/configs/yt8m.py	148
official/projects/yt8m/dataloaders/utils.py	111
official/projects/yt8m/dataloaders/yt8m_input.py	233
official/projects/yt8m/eval_utils/average_precision_calculator.py	100
official/projects/yt8m/eval_utils/eval_util.py	113
official/projects/yt8m/eval_utils/mean_average_precision_calculator.py	31
official/projects/yt8m/experiments/yt8m.yaml	55
official/projects/yt8m/modeling/__init__.py	1
official/projects/yt8m/modeling/yt8m_agg_models.py	57
official/projects/yt8m/modeling/yt8m_model.py	129
official/projects/yt8m/modeling/yt8m_model_utils.py	100
official/projects/yt8m/tasks/__init__.py	1
official/projects/yt8m/tasks/yt8m_task.py	155
official/projects/yt8m/train.py	8