Commit 658c8038 authored by Frederick Liu's avatar Frederick Liu Committed by A. Unique TensorFlower
Browse files

[bigbird] move bigbird from nlp/projects to official/projects

PiperOrigin-RevId: 418851251
parent a7e60974
......@@ -16,9 +16,9 @@
Includes configurations and factory methods.
"""
import dataclasses
from typing import Optional
import dataclasses
import gin
import tensorflow as tf
......@@ -26,7 +26,7 @@ from official.modeling import hyperparams
from official.modeling import tf_utils
from official.nlp.modeling import layers
from official.nlp.modeling import networks
from official.nlp.projects.bigbird import encoder as bigbird_encoder
from official.projects.bigbird import encoder as bigbird_encoder
@dataclasses.dataclass
......
......@@ -20,8 +20,8 @@ import tensorflow as tf
from official.modeling import activations
from official.nlp import modeling
from official.nlp.modeling import layers
from official.nlp.projects.bigbird import recompute_grad
from official.nlp.projects.bigbird import recomputing_dropout
from official.projects.bigbird import recompute_grad
from official.projects.bigbird import recomputing_dropout
_MAX_SEQ_LEN = 4096
......
......@@ -17,7 +17,7 @@
import numpy as np
import tensorflow as tf
from official.nlp.projects.bigbird import encoder
from official.projects.bigbird import encoder
class BigBirdEncoderTest(tf.test.TestCase):
......
......@@ -17,8 +17,8 @@
import numpy as np
import tensorflow as tf
from official.nlp.projects.bigbird import recompute_grad as recompute_grad_lib
from official.nlp.projects.bigbird import stateless_dropout as stateless_dropout_lib
from official.projects.bigbird import recompute_grad as recompute_grad_lib
from official.projects.bigbird import stateless_dropout as stateless_dropout_lib
# Reimplements internal function
......
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