Convention: name mixins mixins

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

View File

@@ -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