[cleanup] assign todos, faster bart-cnn test (#7835)

* 2 beam output

* unassign/remove TODOs

* remove one more
This commit is contained in:
Sam Shleifer
2020-10-16 03:11:18 -04:00
committed by GitHub
parent 7b13bd01df
commit 96e47d9229
8 changed files with 29 additions and 54 deletions

View File

@@ -25,7 +25,6 @@ def test_finetune_trainer():
@slow
def test_finetune_trainer_slow():
# TODO(SS): This will fail on devices with more than 1 GPU.
# There is a missing call to __init__process_group somewhere
output_dir = run_trainer(eval_steps=2, max_len="128", model_name=MARIAN_MODEL, num_train_epochs=3)