Merge pull request #10514 from ZihanWangKi:master

PiperOrigin-RevId: 436548878

Merge pull request #10514 from ZihanWangKi:master
PiperOrigin-RevId: 436548878
36d6f41f · saberkun · bc8b6332 · 259c4347 · 36d6f41f · 36d6f41f
Commit 36d6f41f authored Mar 22, 2022 by saberkun
20 changed files
--- a/official/vision/beta/modeling/factory_test.py
+++ b/official/vision/beta/modeling/factory_test.py
--- a/official/vision/beta/modeling/heads/__init__.py
+++ b/official/vision/beta/modeling/heads/__init__.py
--- a/official/vision/beta/modeling/heads/dense_prediction_heads_test.py
+++ b/official/vision/beta/modeling/heads/dense_prediction_heads_test.py
--- a/official/vision/beta/modeling/heads/instance_heads_test.py
+++ b/official/vision/beta/modeling/heads/instance_heads_test.py
--- a/official/vision/beta/modeling/heads/segmentation_heads_test.py
+++ b/official/vision/beta/modeling/heads/segmentation_heads_test.py
--- a/official/vision/beta/modeling/layers/__init__.py
+++ b/official/vision/beta/modeling/layers/__init__.py
--- a/official/vision/beta/modeling/layers/nn_blocks_3d_test.py
+++ b/official/vision/beta/modeling/layers/nn_blocks_3d_test.py
--- a/official/vision/beta/modeling/layers/nn_blocks_test.py
+++ b/official/vision/beta/modeling/layers/nn_blocks_test.py
--- a/official/vision/beta/modeling/layers/nn_layers_test.py
+++ b/official/vision/beta/modeling/layers/nn_layers_test.py
--- a/official/vision/beta/modeling/maskrcnn_model_test.py
+++ b/official/vision/beta/modeling/maskrcnn_model_test.py
--- a/official/vision/beta/modeling/retinanet_model_test.py
+++ b/official/vision/beta/modeling/retinanet_model_test.py
--- a/official/vision/beta/modeling/segmentation_model_test.py
+++ b/official/vision/beta/modeling/segmentation_model_test.py
--- a/official/vision/beta/modeling/video_classification_model_test.py
+++ b/official/vision/beta/modeling/video_classification_model_test.py
--- a/official/vision/beta/projects/video_ssl/configs/__init__.py
+++ b/official/vision/beta/projects/video_ssl/configs/__init__.py
--- a/official/vision/beta/projects/video_ssl/configs/video_ssl.py
+++ b/official/vision/beta/projects/video_ssl/configs/video_ssl.py
--- a/official/vision/beta/projects/video_ssl/configs/video_ssl_test.py
+++ b/official/vision/beta/projects/video_ssl/configs/video_ssl_test.py
--- a/official/vision/beta/projects/video_ssl/losses/losses.py
+++ b/official/vision/beta/projects/video_ssl/losses/losses.py
--- a/official/vision/beta/projects/video_ssl/ops/video_ssl_preprocess_ops.py
+++ b/official/vision/beta/projects/video_ssl/ops/video_ssl_preprocess_ops.py
--- a/official/vision/beta/projects/video_ssl/tasks/linear_eval.py
+++ b/official/vision/beta/projects/video_ssl/tasks/linear_eval.py
--- a/official/vision/beta/projects/video_ssl/tasks/pretrain.py
+++ b/official/vision/beta/projects/video_ssl/tasks/pretrain.py