Increasing disk size on training job to test MWMS

Lokiiiiii · Lokiiiiii · commit 00f83a5383a3 · 2022-06-14T18:49:54.000-07:00
diff --git a/test/integration/sagemaker/test_multi_worker_mirrored.py b/test/integration/sagemaker/test_multi_worker_mirrored.py
@@ -92,6 +92,7 @@ def test_tf_model_garden(
         },
         max_run=60 * 60 * 12,  # 1 hour
         role="SageMakerRole",
+        volume_size=400,
     )
     estimator.fit(
         inputs="s3://collection-of-ml-datasets/Imagenet/TFRecords/train",

Original file line number	Diff line number	Diff line change
`@@ -92,6 +92,7 @@ def test_tf_model_garden(`
`92`	`92`	`},`
`93`	`93`	`max_run=60 * 60 * 12, # 1 hour`
`94`	`94`	`role="SageMakerRole",`
	`95`	`+ volume_size=400,`
`95`	`96`	`)`
`96`	`97`	`estimator.fit(`
`97`	`98`	`inputs="s3://collection-of-ml-datasets/Imagenet/TFRecords/train",`