Skip to content

Commit ec300ff

Browse files
author
Niklas Palm
committed
refactor: remove unused var and explicit model path
1 parent b1e9ac3 commit ec300ff

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

02_training/training.ipynb

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -103,10 +103,8 @@
103103
"estimator = TensorFlow(\n",
104104
" entry_point=\"train-mobilenet.py\",\n",
105105
" source_dir=\"code\",\n",
106-
" output_path=model_path, \n",
107106
" instance_type=training_instance_type,\n",
108107
" instance_count=training_instance_count, \n",
109-
" distribution=distribution,\n",
110108
" hyperparameters=shared_hyperparameters, \n",
111109
" metric_definitions=metric_definitions, \n",
112110
" role=role,\n",
@@ -197,9 +195,9 @@
197195
}
198196
],
199197
"source": [
200-
"training_job_name = tuner.best_estimator().latest_training_job.name\n",
198+
"best_model_uri = tuner.best_estimator().latest_training_job.describe()['ModelArtifacts']['S3ModelArtifacts']\n",
201199
"\n",
202-
"print(f\"Best model artifact file is uploaded here: {model_path}/{training_job_name}/output ========\")\n"
200+
"print(f\"\\nBest model artifact file is uploaded here: {best_model_uri}\")"
203201
]
204202
}
205203
],
@@ -208,7 +206,7 @@
208206
"kernelspec": {
209207
"display_name": "Python 3 (Data Science)",
210208
"language": "python",
211-
"name": "python3__SAGEMAKER_INTERNAL__arn:aws:sagemaker:us-east-1:081325390199:image/datascience-1.0"
209+
"name": "python3__SAGEMAKER_INTERNAL__arn:aws:sagemaker:eu-west-1:470317259841:image/datascience-1.0"
212210
},
213211
"language_info": {
214212
"codemirror_mode": {

0 commit comments

Comments
 (0)