aws / amazon-sagemaker-examples

Example 📓 Jupyter notebooks that demonstrate how to build, train, and deploy machine learning models using 🧠 Amazon SageMaker.
https://sagemaker-examples.readthedocs.io
Apache License 2.0
10.15k stars 6.78k forks source link

[Example Request] TensorFlow Image Classification Batch Transform #3680

Open dkhundley opened 1 year ago

dkhundley commented 1 year ago

Describe the use case example you want to see In this repo, there is a tutorial on how to properly use transfer learning to train and deploy a TensorFlow image classification mode as a real time endpoint. (Link) It would be awesome to see this further expanded with an example of how to use the trained model in a batch transform job. 😃

How would this example be used? Please describe. After demonstrating how to train the model with custom images with transfer learning, the notebook in the link above would be further expanded to demonstrate how to use that newly trained model in a batch transform context.

Describe which SageMaker services are involved SageMaker notebook SageMaker training job SageMaker batch transform job

Describe what other services (other than SageMaker) are involved N/A

Describe which dataset could be used. Provide its location in s3://sagemaker-sample-files or another source. The same dataset that is used in the example already is totally fine

dkhundley commented 1 year ago

@vivekmadan2 Noticed you were last to adjust this. I'm transparently trying to do this for a hackathon at work that will end in 2 weeks, so your help would be greatly appreciated! 😃

elcronos commented 1 year ago

@dkhundley I also need an example on how to run batch inference on a trained Tensorflow model. Did you find the code you needed? Would you mind to share the code?