Convention: name mixins mixins
This commit is contained in:
parent
055e80cfad
commit
2f32dfd33b
|
@ -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
|
||||
|
|
|
@ -53,7 +53,7 @@ except ImportError:
|
|||
return input
|
||||
|
||||
|
||||
class ModuleUtils:
|
||||
class ModuleUtilsMixin:
|
||||
"""
|
||||
A few utilities for torch.nn.Modules, to be used as a mixin.
|
||||
"""
|
||||
|
@ -66,7 +66,7 @@ class ModuleUtils:
|
|||
return sum(p.numel() for p in params)
|
||||
|
||||
|
||||
class PreTrainedModel(nn.Module, ModuleUtils):
|
||||
class PreTrainedModel(nn.Module, ModuleUtilsMixin):
|
||||
r""" Base class for all models.
|
||||
|
||||
:class:`~transformers.PreTrainedModel` takes care of storing the configuration of the models and handles methods for loading/downloading/saving models
|
||||
|
|
|
@ -592,7 +592,6 @@ class ModelTesterMixin:
|
|||
model(**inputs_dict)
|
||||
|
||||
|
||||
|
||||
global_rng = random.Random()
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue