Fix the TF Trainer gradient accumulation and the TF NER example (#6713)

* Align TF NER example over the PT one * Fix Dataset call * Fix gradient accumulation training * Apply style * Address Sylvain's comments * Address Sylvain's comments * Apply style
2020-08-27 14:45:34 +02:00
parent 41aa2b4ef1
commit 6f289dc97a
4 changed files with 38 additions and 11 deletions
--- a/examples/token-classification/utils_ner.py
+++ b/examples/token-classification/utils_ner.py
@@ -276,7 +276,7 @@ if is_torch_available():
 if is_tf_available():
    import tensorflow as tf

-    class TFNerDataset:
+    class TFTokenClassificationDataset:
        """
        This will be superseded by a framework-agnostic approach
        soon.