Commit f9c5317d authored by Aymeric Augustin's avatar Aymeric Augustin
Browse files

Fix E265 flake8 warning (x1).

parent 28e608a2
...@@ -356,7 +356,7 @@ def train(args, train_dataset, model, tokenizer): ...@@ -356,7 +356,7 @@ def train(args, train_dataset, model, tokenizer):
inputs = { inputs = {
"input_ids": batch[0], "input_ids": batch[0],
"attention_mask": batch[1], "attention_mask": batch[1],
#'token_type_ids': None if args.model_type == 'xlm' else batch[2], # 'token_type_ids': None if args.model_type == 'xlm' else batch[2],
"token_type_ids": batch[2], "token_type_ids": batch[2],
"labels": batch[3], "labels": batch[3],
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment