Update src/run_edu_bert.py
Browse files- src/run_edu_bert.py +1 -1
src/run_edu_bert.py
CHANGED
|
@@ -39,7 +39,7 @@ def main(args):
|
|
| 39 |
if __name__ == "__main__":
|
| 40 |
parser = argparse.ArgumentParser()
|
| 41 |
|
| 42 |
-
parser.add_argument("--model_name", type=str, default="
|
| 43 |
parser.add_argument("--dataset_name", type=str, default="HuggingFaceFW/fineweb")
|
| 44 |
parser.add_argument("--dataset_config", type=str, default="default")
|
| 45 |
parser.add_argument("--output_dataset_name", type=str, default="HuggingFaceFW/fineweb-edu")
|
|
|
|
| 39 |
if __name__ == "__main__":
|
| 40 |
parser = argparse.ArgumentParser()
|
| 41 |
|
| 42 |
+
parser.add_argument("--model_name", type=str, default="HuggingFaceFW/fineweb-edu-classifier")
|
| 43 |
parser.add_argument("--dataset_name", type=str, default="HuggingFaceFW/fineweb")
|
| 44 |
parser.add_argument("--dataset_config", type=str, default="default")
|
| 45 |
parser.add_argument("--output_dataset_name", type=str, default="HuggingFaceFW/fineweb-edu")
|