hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[SGD v2] Reduce time for HF smoke test (#17623)

Browse source 
* reduce

* switch back model

* Update python/ray/util/sgd/v2/BUILD
This commit is contained in:
Amog Kamsetty 2021-08-05 21:04:34 -07:00 • committed by GitHub
parent a06d71477f
commit f0cca063ad
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/ray/util/sgd/v2/BUILD
View file

@ -12,7 +12,7 @@ py_test(
deps = [":sgd_v2_lib"],
args = ["--model_name_or_path=bert-base-cased", "--task_name=mrpc",
"--max_length=32", "--per_device_train_batch_size=64",
"--num_train_epochs=1", "--start_local", "--num_workers=2"]
"--max_train_steps=2", "--start_local", "--num_workers=2"]
)
# --------------------------------------------------------------------