Skip to content

Commit

Permalink
Increase the worker timeout to 1800 seconds
Browse files Browse the repository at this point in the history 
Summary: As the title goes

Reviewed By: dshi7

Differential Revision: D47072520

fbshipit-source-id: 4596c384b853449ed46c0160af09c82fb992684f
  • Loading branch information
@xuzhao9 @facebook-github-bot
xuzhao9 authored and facebook-github-bot committed Jun 27, 2023
1 parent 4e9d86e commit ae3534a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchbenchmark/util/experiment/instantiator.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
from torchbenchmark.util.model import BenchmarkModel
from torchbenchmark import _list_model_paths, load_model_by_name, ModelTask

WORKER_TIMEOUT = 600 # seconds
WORKER_TIMEOUT = 1800 # seconds
BS_FIELD_NAME = "batch_size"

@dataclasses.dataclass
Expand Down

0 comments on commit ae3534a

Please sign in to comment.