Convention: name mixins mixins
This commit is contained in:
@@ -32,7 +32,7 @@ from .modeling_tf_pytorch_utils import load_pytorch_checkpoint_in_tf2_model
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class TFModelUtils:
|
||||
class TFModelUtilsMixin:
|
||||
"""
|
||||
A few utilities for `tf.keras.Model`s, to be used as a mixin.
|
||||
"""
|
||||
@@ -47,7 +47,7 @@ class TFModelUtils:
|
||||
return self.count_params()
|
||||
|
||||
|
||||
class TFPreTrainedModel(tf.keras.Model, TFModelUtils):
|
||||
class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin):
|
||||
r""" Base class for all TF models.
|
||||
|
||||
:class:`~transformers.TFPreTrainedModel` takes care of storing the configuration of the models and handles methods for loading/downloading/saving models
|
||||
|
||||
Reference in New Issue
Block a user