Update pretrain_dataloader.py

tensorflow · Nov 8, 2023 · 19ea3d2 · 19ea3d2
1 parent 8e2ee9c
commit 19ea3d2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/official/nlp/data/pretrain_dataloader.py b/official/nlp/data/pretrain_dataloader.py
@@ -143,7 +143,7 @@ class XLNetPretrainDataConfig(cfg.DataConfig):
 
   Attributes:
     input_path: See base class.
-    global_batch_size: See base calss.
+    global_batch_size: See base class.
     is_training: See base class.
     seq_length: The length of each sequence.
     max_predictions_per_seq: The number of predictions per sequence.