|
131 | 131 | "job_name_prefix = 'rl-deepracer'\n", |
132 | 132 | "\n", |
133 | 133 | "# create unique job name\n", |
134 | | - "job_name = s3_prefix = job_name_prefix + \"-sagemaker-\" + strftime(\"%y%m%d-%H%M%S\", gmtime())\n", |
| 134 | + "tm = gmtime()\n", |
| 135 | + "job_name = s3_prefix = job_name_prefix + \"-sagemaker-\" + strftime(\"%y%m%d-%H%M%S\", tm) #Ensure S3 prefix contains SageMaker\n", |
| 136 | + "s3_prefix_robomaker = job_name_prefix + \"-robomaker-\" + strftime(\"%y%m%d-%H%M%S\", tm) #Ensure that the S3 prefix contains the keyword 'robomaker'\n", |
| 137 | + "\n", |
135 | 138 | "\n", |
136 | 139 | "# Duration of job in seconds (5 hours)\n", |
137 | 140 | "job_duration_in_seconds = 3600 * 5\n", |
|
414 | 417 | "source": [ |
415 | 418 | "RLCOACH_PRESET = \"deepracer\"\n", |
416 | 419 | "\n", |
417 | | - "instance_type = \"ml.c5.4xlarge\"\n", |
| 420 | + "instance_type = \"ml.c4.2xlarge\"\n", |
418 | 421 | "\n", |
419 | 422 | "estimator = RLEstimator(entry_point=\"training_worker.py\",\n", |
420 | 423 | " source_dir='src',\n", |
|
578 | 581 | " maxJobDurationInSeconds=job_duration_in_seconds,\n", |
579 | 582 | " failureBehavior=\"Continue\",\n", |
580 | 583 | " simulationApplications=[simulation_application],\n", |
581 | | - " vpcConfig=vpcConfig\n", |
| 584 | + " vpcConfig=vpcConfig,\n", |
| 585 | + " outputLocation={\"s3Bucket\":s3_bucket, \"s3Prefix\":s3_prefix_robomaker}\n", |
582 | 586 | " )\n", |
583 | 587 | " responses.append(response)\n", |
584 | 588 | "\n", |
|
723 | 727 | " maxJobDurationInSeconds=job_duration_in_seconds,\n", |
724 | 728 | " failureBehavior=\"Continue\",\n", |
725 | 729 | " simulationApplications=[simulation_application],\n", |
726 | | - " vpcConfig=vpcConfig\n", |
| 730 | + " vpcConfig=vpcConfig,\n", |
| 731 | + " outputLocation={\"s3Bucket\":s3_bucket, \"s3Prefix\":s3_prefix_robomaker}\n", |
727 | 732 | " )\n", |
728 | 733 | "print(\"Created the following job:\")\n", |
729 | 734 | "print(\"Job ARN\", response[\"arn\"])" |
|
0 commit comments