Added SageMaker managed training (finetuning) using the same code as used for local training. Added an example of loading a SageMaker trained model into notebook and testing. Factored-out the existing training code to avoid code duplication.
Fully tested on p2.xlarge SageMaker Notebook Instance and SageMaker managed training using p2.xlarge instance.
By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.
Issue #, if available: No associated issue
Description of changes:
Added SageMaker managed training (finetuning) using the same code as used for local training. Added an example of loading a SageMaker trained model into notebook and testing. Factored-out the existing training code to avoid code duplication.
Fully tested on p2.xlarge SageMaker Notebook Instance and SageMaker managed training using p2.xlarge instance.
By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.