Skip to content

Commit

Permalink
update notebook
Browse files Browse the repository at this point in the history
  • Loading branch information
haohanchen-aws committed Oct 5, 2022
1 parent 7104a85 commit b49231f
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -545,7 +545,7 @@
" kwargs[\"security_group_ids\"] = [fsx_security_group_id]\n",
" kwargs[\"subnets\"] = [fsx_subnet]\n",
"\n",
"smp_estimator = HuggingFace(\n",
"smp_estimator = PyTorch(\n",
" entry_point=\"train_gpt_simple.py\",\n",
" source_dir=os.getcwd(),\n",
" role=role,\n",
Expand Down Expand Up @@ -579,8 +579,8 @@
" }\n",
" },\n",
" },\n",
" pytorch_version=\"1.11\",\n",
" transformers_version=\"4.17\",\n",
" pytorch_version=\"1.12\",\n",
" transformers_version=\"4.21\",\n",
" py_version=\"py38\",\n",
" output_path=s3_output_location,\n",
" checkpoint_s3_uri=checkpoint_s3_uri if not use_fsx else None,\n",
Expand Down

0 comments on commit b49231f

Please sign in to comment.