Commit 51364cdf authored by stephenwu's avatar stephenwu
Browse files

fixed style

parent 80a21374
......@@ -121,7 +121,7 @@ class DataProcessor(object):
return lines
@classmethod
def _read_jsonl(self, input_path):
def _read_jsonl(cls, input_file):
"""Reads a json line file."""
with tf.io.gfile.GFile(input_path, "r") as f:
lines = []
......
......@@ -49,7 +49,7 @@ flags.DEFINE_string(
flags.DEFINE_enum(
"classification_task_name", "MNLI", [
"AX", "COLA", "IMDB", "MNLI", "MRPC", "PAWS-X", "QNLI", "QQP", "RTE",
"SST-2", "STS-B", "WNLI", "XNLI", "XTREME-XNLI", "XTREME-PAWS-X", "AX-g",
"SST-2", "STS-B", "WNLI", "XNLI", "XTREME-XNLI", "XTREME-PAWS-X", "AX-g",
"RTE-SuperGLUE"
], "The name of the task to train BERT classifier. The "
"difference between XTREME-XNLI and XNLI is: 1. the format "
......
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