提交 de44470f 编写于 作者: X Xianzhi Du 提交者: A. Unique TensorFlower

Internal change

PiperOrigin-RevId: 423333566
上级 d3a3f143
......@@ -15,4 +15,4 @@
# Lint as: python3
"""Configs package definition."""
from official.vision.beta.projects.vit.configs import image_classification
from official.projects.vit.configs import image_classification
......@@ -26,7 +26,7 @@ from official.modeling import hyperparams
from official.modeling import optimization
from official.vision.beta.configs import common
from official.vision.beta.configs import image_classification as img_cls_cfg
from official.vision.beta.projects.vit.configs import backbones
from official.projects.vit.configs import backbones
from official.vision.beta.tasks import image_classification
DataConfig = img_cls_cfg.DataConfig
......
......@@ -17,9 +17,9 @@
import tensorflow as tf
from official.modeling import activations
from official.projects.vit.modeling import nn_blocks
from official.vision.beta.modeling.backbones import factory
from official.vision.beta.modeling.layers import nn_layers
from official.vision.beta.projects.vit.modeling import nn_blocks
layers = tf.keras.layers
......
......@@ -18,7 +18,7 @@
from absl.testing import parameterized
import tensorflow as tf
from official.vision.beta.projects.vit.modeling import vit
from official.projects.vit.modeling import vit
class VisionTransformerTest(parameterized.TestCase, tf.test.TestCase):
......
......@@ -18,9 +18,9 @@
from absl import app
from official.common import flags as tfm_flags
from official.projects.vit import configs # pylint: disable=unused-import
from official.projects.vit.modeling import vit # pylint: disable=unused-import
from official.vision.beta import train
from official.vision.beta.projects.vit import configs # pylint: disable=unused-import
from official.vision.beta.projects.vit.modeling import vit # pylint: disable=unused-import
if __name__ == '__main__':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册