IBM/MAX-Image-Caption-Generator

Deploy on Amazon SageMaker

oelesinsc24 opened this issue · 1 comments

Do you have any recommendations on deploying this to Amazon SageMaker?

Likewise, can I retrain on my custom dataset?

Thanks in advance

Hi there, thanks for your interest in the Model Asset Exchange!

I'm afraid we cannot assist with SageMaker, being unfamiliar with the details of how that service works. I'd suggest looking into their documentation to see if they support arbitrary Docker container deployments.

For now, we don't support training this model on a custom dataset, please see #46 for details of the underlying training code repo that you can use for that.