Format

0d2c111a · Lysandre · 6f289dc9 · 0d2c111a
Commit 0d2c111a authored Aug 27, 2020 by Lysandre
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/transformers/trainer_tf.py src/transformers/trainer_tf.py +2 -1

No files found.
--- a/src/transformers/trainer_tf.py
+++ b/src/transformers/trainer_tf.py
@@ -629,7 +629,8 @@ class TFTrainer:
                features = {
                    k: tf.concat(
-                        [ft[self.args.train_batch_size // self.args.n_replicas :], reduced_features[k]], axis=0,
+                        [ft[self.args.train_batch_size // self.args.n_replicas :], reduced_features[k]],
+                        axis=0,
                    )
                    for k, ft in features.items()
                }