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

Convention: name mixins mixins

Browse Source
This commit is contained in:
Julien Chaumond
2020-01-11 01:24:29 +00:00
parent 055e80cfad
commit 2f32dfd33b
3 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/modeling_tf_utils.py
View File

@@ -32,7 +32,7 @@ from .modeling_tf_pytorch_utils import load_pytorch_checkpoint_in_tf2_model
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class TFModelUtils: class TFModelUtilsMixin:
""" """
A few utilities for `tf.keras.Model`s, to be used as a mixin. A few utilities for `tf.keras.Model`s, to be used as a mixin.
""" """
@@ -47,7 +47,7 @@ class TFModelUtils:
return self.count_params() return self.count_params()
class TFPreTrainedModel(tf.keras.Model, TFModelUtils): class TFPreTrainedModel(tf.keras.Model, TFModelUtilsMixin):
r""" Base class for all TF models. 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 :class:`~transformers.TFPreTrainedModel` takes care of storing the configuration of the models and handles methods for loading/downloading/saving models

4
src/transformers/modeling_utils.py
View File

@@ -53,7 +53,7 @@ except ImportError:
return input return input
class ModuleUtils: class ModuleUtilsMixin:
""" """
A few utilities for torch.nn.Modules, to be used as a mixin. 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) return sum(p.numel() for p in params)
class PreTrainedModel(nn.Module, ModuleUtils): class PreTrainedModel(nn.Module, ModuleUtilsMixin):
r""" Base class for all models. 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 :class:`~transformers.PreTrainedModel` takes care of storing the configuration of the models and handles methods for loading/downloading/saving models

1
tests/test_modeling_common.py
View File

@@ -592,7 +592,6 @@ class ModelTesterMixin:
model(**inputs_dict) model(**inputs_dict)
global_rng = random.Random() global_rng = random.Random()