Fix E265 flake8 warning (x1).
This commit is contained in:
@@ -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],
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user