Internal change

PiperOrigin-RevId: 395697194
JasperG1998 · Sep 9, 2021 · f8b7b77 · f8b7b77
1 parent 0d9ef8d
commit f8b7b77
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/official/nlp/projects/teams/teams_pretrainer.py b/official/nlp/projects/teams/teams_pretrainer.py
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Trainer network for ELECTRA models."""
+"""Trainer network for TEAMS models."""
 # pylint: disable=g-classes-have-attributes
 
 import tensorflow as tf

diff --git a/official/nlp/projects/teams/teams_task.py b/official/nlp/projects/teams/teams_task.py
@@ -49,7 +49,7 @@ def _get_generator_hidden_layers(discriminator_network, num_hidden_layers,
 
 def _build_pretrainer(
     config: teams.TeamsPretrainerConfig) -> teams_pretrainer.TeamsPretrainer:
-  """Instantiates ElectraPretrainer from the config."""
+  """Instantiates TeamsPretrainer from the config."""
   generator_encoder_cfg = config.generator
   discriminator_encoder_cfg = config.discriminator
   discriminator_network = teams.get_encoder(discriminator_encoder_cfg)