SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Revert "Gradient accumulation for TFTrainer (#9585)"

Browse Source 
This reverts commit 3f40070c88.
This commit is contained in:
Kiyoung Kim
2021-01-15 00:16:39 +09:00
committed by Lysandre
parent 82498cbc37
commit 85788bae5c
1 changed files with 4 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/transformers/trainer_tf.py
View File

@@ -638,9 +638,7 @@ class TFTrainer:
reduced_features = {
k: ft[: self.args.train_batch_size // self.args.n_replicas] for k, ft in features.items()
}
reduced_labels = {
k: lbl[: self.args.train_batch_size // self.args.n_replicas] for k, lbl in labels.items()
}
reduced_labels = labels[: self.args.train_batch_size // self.args.n_replicas]
self.training_step(reduced_features, reduced_labels, nb_instances_in_global_batch)
@@ -652,13 +650,9 @@ class TFTrainer:
for k, ft in features.items()
}
labels = {
k: tf.concat(
[lbl[self.args.train_batch_size // self.args.n_replicas :], reduced_labels[k]],
axis=0,
labels = tf.concat(
[labels[self.args.train_batch_size // self.args.n_replicas :], reduced_labels], axis=0
)
for k, lbl in labels.items()
}
gradients = self.gradient_accumulator.gradients
gradients = [