increase timeout (#17580)

This commit is contained in:
Amog Kamsetty 2021-08-05 02:20:46 -07:00 committed by GitHub
parent 46c1db1aa7
commit d3155bc1a8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -106,7 +106,8 @@ if __name__ == "__main__":
use_gpu=False if args.smoke_test else True,
num_hosts=1 if args.smoke_test else 2,
num_slots=2 if args.smoke_test else 2,
replicate_pem=False)
replicate_pem=False,
timeout_s=300)
transform_train = transforms.Compose([
transforms.RandomCrop(32, padding=4),