remove args
Browse files- run_mlm_flax_stream.py +2 -2
run_mlm_flax_stream.py
CHANGED
@@ -387,9 +387,9 @@ if __name__ == "__main__":
|
|
387 |
if len(sys.argv) == 2 and sys.argv[1].endswith(".json"):
|
388 |
# If we pass only one argument to the script and it's the path to a json file,
|
389 |
# let's parse it to get our arguments.
|
390 |
-
model_args, data_args, training_args
|
391 |
else:
|
392 |
-
model_args, data_args, training_args
|
393 |
|
394 |
if (
|
395 |
os.path.exists(training_args.output_dir)
|
|
|
387 |
if len(sys.argv) == 2 and sys.argv[1].endswith(".json"):
|
388 |
# If we pass only one argument to the script and it's the path to a json file,
|
389 |
# let's parse it to get our arguments.
|
390 |
+
model_args, data_args, training_args = parser.parse_json_file(json_file=os.path.abspath(sys.argv[1]))
|
391 |
else:
|
392 |
+
model_args, data_args, training_args = parser.parse_args_into_dataclasses()
|
393 |
|
394 |
if (
|
395 |
os.path.exists(training_args.output_dir)
|