hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-04 17:41:43 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[tests] Remove unnecessary sleep time from pipelined ingest tests #28182

Browse source 
Signed-off-by: Stephanie Wang <swang@cs.berkeley.edu>
This commit is contained in:
Stephanie Wang 2022-08-31 20:43:58 -04:00 • committed by GitHub
parent 5cec2492bb
commit 213e24cafd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
release/nightly_tests/dataset/pipelined_training.py
View file

@ -44,7 +44,6 @@ parser.add_argument(
help=("how many batches to wait before logging training " "status"), help=("how many batches to wait before logging training " "status"),
) )
parser.add_argument("--num-workers", type=int, default=16) parser.add_argument("--num-workers", type=int, default=16)
parser.add_argument("--mock-train-step-time", type=float, default=1.0)
parser.add_argument("--num-files", type=int, default=30) parser.add_argument("--num-files", type=int, default=30)
parser.add_argument("--num-windows", type=int, default=1) parser.add_argument("--num-windows", type=int, default=1)
parser.add_argument("--manual-windows", type=bool, default=False) parser.add_argument("--manual-windows", type=bool, default=False)
@ -137,7 +136,6 @@ def train_main(args, splits):
f"Processing batch {batch_idx} in epoch {epoch} on worker " f"Processing batch {batch_idx} in epoch {epoch} on worker "
f"{rank}." f"{rank}."
) )
time.sleep(args.mock_train_step_time)
loss = loss_function(batch_pred, target, delta=60) loss = loss_function(batch_pred, target, delta=60)
loss.mean().backward() loss.mean().backward()
for opt in optimizers: for opt in optimizers: