Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
aedb9802
Commit
aedb9802
authored
Nov 15, 2019
by
Hongkun Yu
Committed by
A. Unique TensorFlower
Nov 15, 2019
Browse files
Move XLNet config and modeling code to xlnet folder.
PiperOrigin-RevId: 280720540
parent
98f10421
Changes
7
Show whitespace changes
Inline
Side-by-side
Showing
7 changed files
with
10 additions
and
10 deletions
+10
-10
official/nlp/xlnet/run_classifier.py
official/nlp/xlnet/run_classifier.py
+2
-2
official/nlp/xlnet/run_pretrain.py
official/nlp/xlnet/run_pretrain.py
+2
-2
official/nlp/xlnet/run_squad.py
official/nlp/xlnet/run_squad.py
+2
-2
official/nlp/xlnet/training_utils.py
official/nlp/xlnet/training_utils.py
+3
-3
official/nlp/xlnet/xlnet_config.py
official/nlp/xlnet/xlnet_config.py
+0
-0
official/nlp/xlnet/xlnet_modeling.py
official/nlp/xlnet/xlnet_modeling.py
+0
-0
official/nlp/xlnet/xlnet_modeling_test.py
official/nlp/xlnet/xlnet_modeling_test.py
+1
-1
No files found.
official/nlp/xlnet/run_classifier.py
View file @
aedb9802
...
...
@@ -27,12 +27,12 @@ from absl import logging
import
numpy
as
np
import
tensorflow
as
tf
# pylint: disable=unused-import
from
official.nlp
import
xlnet_config
from
official.nlp
import
xlnet_modeling
as
modeling
from
official.nlp.xlnet
import
common_flags
from
official.nlp.xlnet
import
data_utils
from
official.nlp.xlnet
import
optimization
from
official.nlp.xlnet
import
training_utils
from
official.nlp.xlnet
import
xlnet_config
from
official.nlp.xlnet
import
xlnet_modeling
as
modeling
from
official.utils.misc
import
tpu_lib
flags
.
DEFINE_integer
(
"n_class"
,
default
=
2
,
help
=
"Number of classes."
)
...
...
official/nlp/xlnet/run_pretrain.py
View file @
aedb9802
...
...
@@ -27,12 +27,12 @@ from absl import flags
from
absl
import
logging
import
tensorflow
as
tf
# pylint: disable=unused-import
from
official.nlp
import
xlnet_config
from
official.nlp
import
xlnet_modeling
as
modeling
from
official.nlp.xlnet
import
common_flags
from
official.nlp.xlnet
import
data_utils
from
official.nlp.xlnet
import
optimization
from
official.nlp.xlnet
import
training_utils
from
official.nlp.xlnet
import
xlnet_config
from
official.nlp.xlnet
import
xlnet_modeling
as
modeling
from
official.utils.misc
import
tpu_lib
flags
.
DEFINE_integer
(
...
...
official/nlp/xlnet/run_squad.py
View file @
aedb9802
...
...
@@ -31,13 +31,13 @@ from absl import logging
import
tensorflow
as
tf
# pylint: disable=unused-import
import
sentencepiece
as
spm
from
official.nlp
import
xlnet_config
from
official.nlp
import
xlnet_modeling
as
modeling
from
official.nlp.xlnet
import
common_flags
from
official.nlp.xlnet
import
data_utils
from
official.nlp.xlnet
import
optimization
from
official.nlp.xlnet
import
squad_utils
from
official.nlp.xlnet
import
training_utils
from
official.nlp.xlnet
import
xlnet_config
from
official.nlp.xlnet
import
xlnet_modeling
as
modeling
from
official.utils.misc
import
tpu_lib
flags
.
DEFINE_string
(
...
...
official/nlp/xlnet/training_utils.py
View file @
aedb9802
...
...
@@ -13,7 +13,6 @@
# limitations under the License.
# ==============================================================================
"""XLNet training utils."""
from
__future__
import
absolute_import
from
__future__
import
division
# from __future__ import google_type_annotations
...
...
@@ -27,10 +26,11 @@ from absl import logging
# pytype: disable=attribute-error
# pylint: disable=g-bare-generic,unused-import
import
tensorflow
as
tf
from
typing
import
Any
,
Callable
,
Dict
,
Text
,
Optional
from
official.modeling
import
model_training_utils
from
official.nlp.xlnet
import
data_utils
from
official.nlp
import
xlnet_modeling
as
modeling
from
typing
import
Any
,
Callable
,
Dict
,
Text
,
Optional
from
official.nlp.xlnet
import
xlnet_modeling
as
modeling
_MIN_SUMMARY_STEPS
=
10
...
...
official/nlp/xlnet_config.py
→
official/nlp/xlnet
/xlnet
_config.py
View file @
aedb9802
File moved
official/nlp/xlnet_modeling.py
→
official/nlp/xlnet
/xlnet
_modeling.py
View file @
aedb9802
File moved
official/nlp/xlnet_modeling_test.py
→
official/nlp/xlnet
/xlnet
_modeling_test.py
View file @
aedb9802
...
...
@@ -20,7 +20,7 @@ from absl import logging
import
numpy
as
np
import
tensorflow
as
tf
from
official.nlp
import
xlnet_modeling
from
official.nlp
.xlnet
import
xlnet_modeling
class
PositionalEmbeddingLayerTest
(
tf
.
test
.
TestCase
):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment