SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Samell fixed in tf template (#7044)

Browse Source
This commit is contained in:
Sylvain Gugger
2020-09-10 10:36:02 -04:00
committed by GitHub
parent 63e539459d
commit d1691d90e5
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
templates/adding_a_new_model/modeling_tf_xxx.py
View File

@@ -19,9 +19,6 @@
# In this template, replace all the XXX (various casings) with your model name # In this template, replace all the XXX (various casings) with your model name
#################################################### ####################################################
import logging
import tensorflow as tf import tensorflow as tf
from .configuration_xxx import XxxConfig from .configuration_xxx import XxxConfig
@@ -47,12 +44,14 @@ from .modeling_tf_utils import (
TFSequenceClassificationLoss, TFSequenceClassificationLoss,
TFTokenClassificationLoss, TFTokenClassificationLoss,
get_initializer, get_initializer,
keras_serializable,
shape_list, shape_list,
) )
from .tokenization_utils import BatchEncoding from .tokenization_utils import BatchEncoding
from .utils import logging
logger = logging.getLogger(__name__) logger = logging.get_logger(__name__)
_CONFIG_FOR_DOC = "XXXConfig" _CONFIG_FOR_DOC = "XXXConfig"
_TOKENIZER_FOR_DOC = "XxxTokenizer" _TOKENIZER_FOR_DOC = "XxxTokenizer"
@@ -115,6 +114,7 @@ class TFXxxLayer(tf.keras.layers.Layer):
# The full model without a specific pretrained or finetuning head is # The full model without a specific pretrained or finetuning head is
# provided as a tf.keras.layers.Layer usually called "TFXxxMainLayer" # provided as a tf.keras.layers.Layer usually called "TFXxxMainLayer"
#################################################### ####################################################
@keras_serializable
class TFXxxMainLayer(tf.keras.layers.Layer): class TFXxxMainLayer(tf.keras.layers.Layer):
def __init__(self, config, **kwargs): def __init__(self, config, **kwargs):
super().__init__(**kwargs) super().__init__(**kwargs)