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"),
)
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-windows", type=int, default=1)
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"{rank}."
)
time.sleep(args.mock_train_step_time)
loss = loss_function(batch_pred, target, delta=60)
loss.mean().backward()
for opt in optimizers: