Commit c57e975a authored by saberkun's avatar saberkun
Browse files

Merge pull request #10338 from srihari-humbarwadi:readme

PiperOrigin-RevId: 413033276
parents 7fb4f3cd acf4156e
......@@ -28,7 +28,7 @@ from official.nlp.configs import encoders
from official.nlp.data import pretrain_dataloader
from official.nlp.modeling import layers
from official.nlp.modeling import models
from official.nlp.projects.mobilebert import distillation
from official.projects.mobilebert import distillation
class DistillationTest(tf.test.TestCase, parameterized.TestCase):
......
......@@ -18,7 +18,7 @@ from absl import flags
from absl import logging
import tensorflow as tf
from official.nlp.projects.mobilebert import model_utils
from official.projects.mobilebert import model_utils
FLAGS = flags.FLAGS
......
......@@ -31,7 +31,7 @@ from official.modeling import performance
from official.modeling.fast_training.progressive import train_lib
from official.modeling.fast_training.progressive import trainer as prog_trainer_lib
from official.nlp.data import pretrain_dataloader
from official.nlp.projects.mobilebert import distillation
from official.projects.mobilebert import distillation
FLAGS = flags.FLAGS
......
......@@ -21,7 +21,7 @@ from absl import logging
import numpy as np
import tensorflow.compat.v1 as tf
from official.nlp.projects.mobilebert import model_utils
from official.projects.mobilebert import model_utils
FLAGS = flags.FLAGS
......
......@@ -12,8 +12,3 @@
# See the License for the specific language governing permissions and
# limitations under the License.
"""Keras-CV layers package definition."""
from official.vision.keras_cv.ops.anchor_generator import AnchorGenerator
from official.vision.keras_cv.ops.box_matcher import BoxMatcher
from official.vision.keras_cv.ops.iou_similarity import IouSimilarity
from official.vision.keras_cv.ops.target_gather import TargetGather
......@@ -14,9 +14,8 @@
"""Common NHNet/Bert2Bert configuration."""
from typing import List, Text
import dataclasses
from typing import List, Text
from official.modeling.hyperparams import base_config
......
......@@ -15,7 +15,7 @@
"""Tests for configs."""
import tensorflow as tf
from official.nlp.nhnet import configs
from official.projects.nhnet import configs
BERT2BERT_CONFIG = {
"vocab_size": 30522,
......
......@@ -12,14 +12,14 @@
# See the License for the specific language governing permissions and
# limitations under the License.
"""Tests for nlp.nhnet.decoder."""
"""Tests for projects.nhnet.decoder."""
import numpy as np
import tensorflow as tf
from official.nlp.modeling import layers
from official.nlp.nhnet import configs
from official.nlp.nhnet import decoder
from official.nlp.nhnet import utils
from official.projects.nhnet import configs
from official.projects.nhnet import decoder
from official.projects.nhnet import utils
class DecoderTest(tf.test.TestCase):
......
......@@ -15,15 +15,17 @@
"""Evaluation for Bert2Bert."""
import os
# Import libraries
from absl import logging
import numpy as np
import tensorflow as tf
from official.nlp.nhnet import input_pipeline
from official.nlp.nhnet import models
from official.nlp.transformer import metrics as metrics_v2
from official.nlp.transformer.utils import metrics
from official.projects.nhnet import input_pipeline
from official.projects.nhnet import models
def rouge_l_fscore(logits, labels):
......
......@@ -24,9 +24,9 @@ from official.modeling.hyperparams import params_dict
from official.nlp.modeling import networks
from official.nlp.modeling.layers import multi_channel_attention
from official.nlp.modeling.ops import beam_search
from official.nlp.nhnet import configs
from official.nlp.nhnet import decoder
from official.nlp.nhnet import utils
from official.projects.nhnet import configs
from official.projects.nhnet import decoder
from official.projects.nhnet import utils
def embedding_linear(embedding_matrix, x):
......
......@@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
"""Tests for nlp.nhnet.models."""
"""Tests for projects.nhnet.models."""
import os
......@@ -25,9 +25,9 @@ import tensorflow as tf
from tensorflow.python.distribute import combinations
from tensorflow.python.distribute import strategy_combinations
# pylint: enable=g-direct-tensorflow-import
from official.nlp.nhnet import configs
from official.nlp.nhnet import models
from official.nlp.nhnet import utils
from official.projects.nhnet import configs
from official.projects.nhnet import models
from official.projects.nhnet import utils
def all_strategy_combinations():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment