Skip to content

Commit 28213b2

Browse files
yeqinglitensorflower-gardener
authored andcommitted
Internal change
PiperOrigin-RevId: 416593512
1 parent 00cc241 commit 28213b2

File tree

11 files changed

+16
-17
lines changed

11 files changed

+16
-17
lines changed
File renamed without changes.

official/vision/beta/projects/assemblenet/configs/assemblenet.py renamed to official/projects/assemblenet/configs/assemblenet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,14 +35,14 @@
3535
The structure_weights specify the learned connection weights.
3636
"""
3737
import dataclasses
38-
from typing import List, Tuple, Optional
38+
from typing import List, Optional, Tuple
3939

4040
from official.core import config_definitions as cfg
4141
from official.core import exp_factory
4242
from official.modeling import hyperparams
4343
from official.vision.beta.configs import backbones_3d
4444
from official.vision.beta.configs import common
45-
from official.vision.beta.configs import video_classification
45+
from official.vision.beta.configs.google import video_classification
4646

4747

4848
@dataclasses.dataclass

official/vision/beta/projects/assemblenet/configs/assemblenet_test.py renamed to official/projects/assemblenet/configs/assemblenet_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@
1717
import tensorflow as tf
1818
from official.core import config_definitions as cfg
1919
from official.core import exp_factory
20+
from official.projects.assemblenet.configs import assemblenet
2021
from official.vision.beta.configs import video_classification as exp_cfg
21-
from official.vision.beta.projects.assemblenet.configs import assemblenet
2222

2323

2424
class AssemblenetTest(tf.test.TestCase, parameterized.TestCase):

official/vision/beta/projects/assemblenet/modeling/assemblenet.py renamed to official/projects/assemblenet/modeling/assemblenet.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -48,17 +48,17 @@
4848

4949
import functools
5050
import math
51-
from typing import Any, Mapping, List, Callable, Optional
51+
from typing import Any, Callable, List, Mapping, Optional
5252

5353
from absl import logging
5454
import numpy as np
5555
import tensorflow as tf
5656

5757
from official.modeling import hyperparams
58+
from official.projects.assemblenet.configs import assemblenet as cfg
59+
from official.projects.assemblenet.modeling import rep_flow_2d_layer as rf
5860
from official.vision.beta.modeling import factory_3d as model_factory
5961
from official.vision.beta.modeling.backbones import factory as backbone_factory
60-
from official.vision.beta.projects.assemblenet.configs import assemblenet as cfg
61-
from official.vision.beta.projects.assemblenet.modeling import rep_flow_2d_layer as rf
6262

6363
layers = tf.keras.layers
6464
intermediate_channel_size = [64, 128, 256, 512]

official/vision/beta/projects/assemblenet/modeling/assemblenet_plus.py renamed to official/projects/assemblenet/modeling/assemblenet_plus.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -54,18 +54,18 @@
5454
"""
5555

5656
import functools
57-
from typing import Any, Dict, Mapping, List, Optional
57+
from typing import Any, Dict, List, Mapping, Optional
5858

5959
from absl import logging
6060
import numpy as np
6161
import tensorflow as tf
6262

6363
from official.modeling import hyperparams
64+
from official.projects.assemblenet.configs import assemblenet as cfg
65+
from official.projects.assemblenet.modeling import assemblenet as asn
66+
from official.projects.assemblenet.modeling import rep_flow_2d_layer as rf
6467
from official.vision.beta.modeling import factory_3d as model_factory
6568
from official.vision.beta.modeling.backbones import factory as backbone_factory
66-
from official.vision.beta.projects.assemblenet.configs import assemblenet as cfg
67-
from official.vision.beta.projects.assemblenet.modeling import assemblenet as asn
68-
from official.vision.beta.projects.assemblenet.modeling import rep_flow_2d_layer as rf
6969

7070
layers = tf.keras.layers
7171

official/vision/beta/projects/assemblenet/modeling/assemblenet_plus_test.py renamed to official/projects/assemblenet/modeling/assemblenet_plus_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@
1818
import numpy as np
1919
import tensorflow as tf
2020

21-
from official.vision.beta.projects.assemblenet.configs import assemblenet as asn_config
22-
from official.vision.beta.projects.assemblenet.modeling import assemblenet_plus as asnp
21+
from official.projects.assemblenet.configs import assemblenet as asn_config
22+
from official.projects.assemblenet.modeling import assemblenet_plus as asnp
2323

2424

2525
class AssembleNetPlusTest(parameterized.TestCase, tf.test.TestCase):

official/vision/beta/projects/assemblenet/train.py renamed to official/projects/assemblenet/train.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,9 @@
3939
from official.core import train_utils
4040
from official.modeling import performance
4141
# pylint: disable=unused-import
42-
from official.vision.beta.projects.assemblenet.configs import assemblenet as asn_configs
43-
from official.vision.beta.projects.assemblenet.modeling import assemblenet as asn
44-
from official.vision.beta.projects.assemblenet.modeling import assemblenet_plus as asnp
42+
from official.projects.assemblenet.configs import assemblenet as asn_configs
43+
from official.projects.assemblenet.modeling import assemblenet as asn
44+
from official.projects.assemblenet.modeling import assemblenet_plus as asnp
4545
# pylint: enable=unused-import
4646

4747
FLAGS = flags.FLAGS

0 commit comments

Comments
 (0)